语义分割学习笔记2——Jetson Xavier安装pytorch

语义分割学习笔记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

语义分割学习笔记2——Jetson Xavier安装pytorch

测试是否运行成功

python3
import torch

语义分割学习笔记2——Jetson Xavier安装pytorch

二、安装torchvision

pytorch和torchvision版本对应关系
以下版本截止2021年3月24日
语义分割学习笔记2——Jetson Xavier安装pytorch
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

语义分割学习笔记2——Jetson Xavier安装pytorch

参考:
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

语义分割学习笔记2——Jetson Xavier安装pytorch

大功告成!

上一篇:玩转NVIDIA Jetson AGX Xavier(2)--- Jetson系列产品简介


下一篇:NVIDIA JETSON TX2 开发板刷机