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

輸出：