How to download torch 1.7.1 or 2.1.0 with cuda >= 11.0

Robotics & Edge Computing Jetson & Embedded Systems Jetson AGX Xavier
,
1

$ pip3 install torch==1.7.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
Collecting torch==1.7.1+cu111
Could not find a version that satisfies the requirement torch==1.7.1+cu111 (from versions: 1.7.1, 1.9.0)
Hi all, I have tried installing both 1.7.1 and 2.1.0 version of pytorch which comes with cuda >= 11.0
Please provide me the right commands to install a pytorch, torchvision compatible with cuda > 11.0

But I got below errors
No matching distribution found for torch==1.7.1+cu111

pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu118
Collecting torch==2.1.0
Could not find a version that satisfies the requirement torch==2.1.0 (from versions: )
No matching distribution found for torch==2.1.0

Thanks and Regards

Nagaraj Trivedi

3

Hi,

The official package doesn’t support Jetson which is an ARM system.
Please check the below document to use our prebuilt instead:

Thanks.

4

Hi, thank you. I will try it and let you the result.

Thanks and Regards

Nagaraj Trivedi

6

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.