由于TX2是arm架构,无法直接通过官方的方法——pip或conda等安装,需要自行编译github上的软件包才能使用。
注:本文略过TX2上的CUDA,Cudnn以及opencv的安装,这些部分的安装大家可以自行百度。
现在正式开始编译pytorch吧:
1.打开终端,输入以下命令下载git包
git clone https://github.com/pytorch/pytorch.git
下载完后,大家会看到在home下有了pytorch文件夹。
2.切分版本
cd到home下的pytorch文件夹,进行版本切分,我要使用pytorch0.3.1版本,所以依次输入以下命令进行切分:
cd '/home/nvidia/pytorch'
git checkout -b v0.3.1 v0.3.1
3.更新submodule
此步不进行,在install的时候会报Could not find /home/nvidia/pytorch/torch/lib/gloo/CMakeLists.txt错误
在pytorch文件夹路径下,在终端输入以下命令进行更新:
git submodule update --init
4.正式编译安装
还是在当前pytorch文件夹路径下,输入以下命令进行编译安装:
sudo python setup.py install
切记一定要带上sudo,不然编译很久很久要完成后,你会发现报Permission denied的错误。
(编译时间很长,大家请耐心等待,睡一觉就好了)
5.测试
在python下输入import torch,没有显示错误则说明编译安装成功,如下图:
查看pytorch版本的方法
import torch
print(torch.__version__) #注意是双下划线