语义分割学习笔记2——Jetson Xavier安装pytorch
我的环境说明：
| 名称 | 版本 |
|---|---|
| 硬件： | Nvidia Jetson Xavier AGX |
| CUDA版本： | 10.2 |
| Jetpack版本： | JP 4.5 |
| python版本： | python2.7 python3.6 |
| deepstream-app version | 5.0.0 |
| TensorRT Version: | 7.1 |
| cuDNN Version: | 8.0 |
一、whl文件安装pytorch
在英伟达官方提供的Jetson Zoo处下载你系统对应的英伟达编译好的arm版本，这里和平常linux系统下安装的whl包不同，Jetson Xavier是arm环境不是x86。
例如我下载的文件是
然后按照下面四个步骤安装pytorch，这里注意，不是直接运行
pip install torch-1.8.0-cp36-cp36m-linux_aarch64.whl
而是需要运行下面四个步骤才能成功。
安装 OpenBLAS and OpenMPI
sudo apt-get install libopenblas-base libopenmpi-dev
sudo apt-get install python3-pip
pip3 install Cython
pip3 install numpy torch-1.8.0-cp36-cp36m-linux_aarch64.whl
测试是否运行成功
python3
import torch
二、安装torchvision
pytorch和torchvision版本对应关系
以下版本截止2021年3月24日
pytorch1.8.0对应的torchvision版本为0.9.0
pip3 install torchvison==0.9.0
三、测试是否安装成功
import torch
import torchvision
print(torch.__version__)
print(torch.version.cuda)
print(torchvision.__version__)
运行
python3 test_pytorch.py
大功告成!
