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
>>>  
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章