本文主要是介绍语义分割学习笔记2——Jetson Xavier安装pytorch，对大家解决编程问题具有一定的参考价值，需要的程序猿们随着小编来一起学习吧！

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

• 一、whl文件安装pytorch
• 二、安装torchvision
• 三、测试是否安装成功

运行语义分割的代码，需要先安装pytorch，下面是我在Xavier上安装语义分割的步骤。
我的环境说明：

一、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

大功告成!

这篇关于语义分割学习笔记2——Jetson Xavier安装pytorch的文章就介绍到这儿，希望我们推荐的文章对大家有所帮助，也希望大家多多支持为之网！