Jetson TX2安装编译pytorch

工作需要将deepsort移植到TX2上面，从网上搜集了教程。

安装教程：

1. 最好是重新安装一个jetpack系统，3.2及以上都可以。

2. 首先安装环境依赖：因为是在python3下面安装pytorch，所以使用pip3。

sudo apt install libopenblas-dev libatlas-base-dev liblapack-dev
# For OpenCV
sudo apt install liblapacke-dev checkinstall
sudo apt-get install python3-pip
sudo apt-get install python3-dev

# 安装python库
sudo apt-get install python3-dev python3-pip libxml2-dev libxslt1-dev zlib1g-dev libffi-dev libssl-dev
sudo apt-get install gfortran  
sudo pip3 install numpy scipy 
sudo pip3 install pyyaml
sudo pip3 install scikit-build
sudo apt-get -y install cmake
sudo pip3 install cffi

3. 添加CUDNN路径：我们刷机之后CUDNN和CUDA都是安装好的，但是我们需要手动将CUDNN添加到环境中。

sudo gedit ~/.bashrc
export CUDNN_LIB_DIR=/usr/lib/aarch64-linux-gnu
export CUDNN_INCLUDE_DIR=/usr/include
source ~/.bashrc

4. 下载pytorch安装包，并且安装pip3所需要的其他安装包。

git clone http://github.com/pytorch/pytorch
cd pytorch
sudo pip3 install -U setuptools
sudo pip3 install -r requirements.txt

5. 选择pytorch要编译的版本。

# 这里选择1.0.0版本进行编译
git checkout tags/v1.0.0 -b build  #这一句话很重要是选择你要编译的pytorch版本
git submodule update --init --recursive

这里需要注意的是当你执行git submodule update --init --recursive  这句命令时候需要你输入你的github的用户名和密码来下载 https://github.com/NervanaSystems/nervanagpu.git 这个安装包，但是这个安装包已经丢失下载不到了。所以我们需要将这个包在git上删除掉，然后再运行git submodule update --init --recursive。

命令行如下：删除nervanagpu这个包。

git clean -xdf
git submodule deinit -f .
git submodule sync
git submodule update --init --recursive

6. 调整TX2的工作效率

sudo nvpmodel -m 0         # 切换工作模式到最大
cd  /usr/bin/
sudo ./jetson_clocks       # 强制开启风扇最大转速

7. 开始编译pytorch：这两句命令编译时间很长一般在4小时左右。

sudo python3 setup.py build
​
NO_SYSTEM_NCCL=1 DEBUG=1 sudo -E python3 setup.py install

8. 后续操作

sudo apt clean
sudo apt-get install libjpeg-dev zlib1g-dev

git clone https://github.com/python-pillow/Pillow.git
cd Pillow/
sudo python3 setup.py install
sudo pip3 install pandas # ~20-30 min
sudo pip3 install Cython
sudo pip3 install scikit-image

sudo pip3 --no-cache-dir install torchvision

9. 验证安装

$ sudo python3   #运行python3的时候加上管理员权限，如果不加sudo，程序会报CUDA错误

>>> import torch
>>>