TX2 (jetpack4.2)安装pytorch详细教程

TX2 刷完jetpack4.2固件后安装的ubuntu系统同时存在两个版本的python，一个是默认的python2.7,另一个是python3.6版本，我安装pytorch1.1.0使用的是2.7版本。

1、安装pytorch

（1）、将TX2开启最大工作模式

sudo nvpmodel -m 0 
cd usr/bin
sudo ./jetson_clocks 

（2）、安装pip

sudo apt-get install python-pip

pip install --upgrade pip    #更新pip版本

（3）、安装必要的依赖

sudo apt-get install libopenblas-base libopenmpi-dev

（4）、安装pytorch

TX2安装Pytorch最简单的方法就是下载nvidia官网提供的对应版本的pytorch的whl文件和教程进行安装，nvidia官方pythorch下载地址。本次选择的是安装的python2.7版本的pytorch1.1.0版本。

 

安装的方式有很多种：

第一种方法：通过wget下载whl文件有时会比较慢甚至下载失败，我是在windows环境下用利用KEXUE上网下载，速度会比较快，下载好后将文件放在home目录下然后执行后面2步。

wget https://nvidia.box.com/shared/static/1v2cc4ro6zvsbu0p8h6qcuaqco1qcsif.whl -O torch-1.1.0-cp27-cp27mu-linux_aarch64.whl
sudo apt-get install libopenblas-base libopenmpi-dev 
pip install future torch-1.1.0-cp27-cp27mu-linux_aarch64.whl

安装完成后可以在python环境下通过import torch进行验证

　　

2 安装Torchvision v0.3.0 

通过以下命令即可完成安装

sudo apt-get install libjpeg-dev zlib1g-dev
#实测用手机热点下载会比较顺利
git clone --branch v0.3.0 https://github.com/pytorch/vision torchvision
cd torchvision
#这一步花费的时间较长
sudo python setup.py install
cd ../
pip install 'pillow<7' # always needed for Python 2.7, not needed torchvision v0.5.0+ with Python 3.6

进行到sudo python setup.py install这一步时，报错如下： 

　

我也不知道啥原因我就编辑了bashrc文件加入了CUDA和CUDNN路径：

sudo gedit ~/.bashrc
export CUDNN_LIB_DIR=/usr/lib/aarch64-linux-gnu
export CUDNN_INCLUDE_DIR=/usr/include
export CUDA_ROOT="/usr/local/cuda-10.0/"
export LD_LIBRARY_PATH="/usr/local/cuda-10.0/lib64/:$LD_LIBRARY_PATH"
source ~/.bashrc

也可能是是安装pillow的过程失败。通过下面的语句可实现加速安装pillow，便可解决问题，然后接着执行后面的语句。

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple pillow==4.1.1

3 安装验证

通过一下命令进行安装验证

python -c "import torch ; print(torch.__version__)"

安装正确的话，结果如：

4 测试

新建文件

gedit test.py

内容如下：

import torch
print(torch.cuda.is_available())
a = torch.cuda.FloatTensor(2)
print(a)
b = torch.randn(2).cuda()
print(b)
c = a + b
print(c)

执行结果：

python  test.py

输出：