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