语义分割学习笔记2——Jetson Xavier安装pytorch
运行语义分割的代码,需要先安装pytorch,下面是我在Xavier上安装语义分割的步骤。
我的环境说明:
名称 | 版本 |
---|---|
硬件: | 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.6.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.6.0-cp36-cp36m-linux_aarch64.whl
测试是否运行成功
python3
import torch
二、安装torchvision
pytorch和torchvision版本对应关系
以下版本截止2021年3月24日
pytorch1.6.0对应的torchvision版本为0.7.0
git clone -b v0.7.0 https://github.com/pytorch/vision
cd vision
sudo python3 setup.py install
参考:
https://blog.csdn.net/u013595260/article/details/107180928/
三、测试是否安装成功
import torch
import torchvision
print(torch.__version__)
print(torch.version.cuda)
print(torchvision.__version__)
运行
python3 test_pytorch.py
大功告成!