pip install problem: Could not find a version that satisfies the requirement cupy-cuda102

I’m having the exact same error as received in #2490.

System details:

• Machine: Jetson Nano
• OS: Ubuntu 18.04.5
• Python: 3.6.9
• Pip: 21.0.1
• CUDA: 10.2

Running the command python3 -m pip install cupy-cuda102 fails. Am I missing a package or something?

Here is the log:

$ python3 -m pip install cupy-cuda102
Collecting cupy-cuda102
  Could not find a version that satisfies the requirement cupy-cuda102 (from versions: )
No matching distribution found for cupy-cuda102

[leofang]

Oh, that’s probably the issue, I’m using the Jetson Nano, which I believe is ARM.

I see, I think we should have documented it 😃

Perhaps it’s worth keeping track in #3196.

[kmaehashi]

Oh, that’s probably the issue, I’m using the Jetson Nano, which I believe is ARM.

I see, I think we should have documented it 😃

Should CuPy still work for CUDA 10.2 if I install using python -m pip install cupy?

Yes, it should work. You can use CUPY_NVCC_GENERATE_CODE environment variable (https://github.com/rapidsai/cusignal/issues/304#issuecomment-758014463) to reduce build time.