在 NVIDIA Jetson上安装GPU版本的pytorch和torchvision
这些pytorch包仅支持python3.6,使用conda直接创建python3.6时会报错,修改命令为。参考链接【3】处下载torch安装包。来正确创建python3.6环境。
注意:pytorch和torchvision都不能直接使用pip install进行安装,这样安装的版本是cpu版本,需要去参考链接【3】处下载torch安装包,下载完成后离线安装。
pip install path/to/your/whl_file.whl
Jetpack版本与pytorch的部分对应表,如下图,完整对应请查看参考链接【3】:
版本对应表 |
|
torch |
torchvision |
1.13.1 |
0.14.1 |
1.13.0 |
0.14.0 |
1.12.0 |
0.13.0 |
1.11.0 |
0.12.0 |
1.10.0 |
0.11.1 |
1.9.0 |
0.10.0 |
1.8.0 |
0.9.0 |
此处有torch1.8-1.10的whl文件以及对应的torchvision文件:
链接:https://pan.baidu.com/s/1YD1y_-v6YppjaULUhZ9rPQ?pwd=zp1h
提取码:zp1h
点此下载
安装时会自动安装numpy,特别注意,torch-1.10对应的numpy版本一定要1.19.3的,默认安装的1.19.5的会报错
这些pytorch包仅支持python3.6,使用conda直接创建python3.6时会报错,修改命令为
conda create -n <env_name> python=3.6 -c conda-forge
来正确创建python3.6环境
参考链接:
【2】NVIDIA JETSON XAVIER NX (四)安装Pytorch和torchvision_jetson安装pytorch-CSDN博客
【3】PyTorch for Jetson - Jetson & Embedded Systems / Jetson Nano - NVIDIA Developer Forums
更多推荐
所有评论(0)