目錄

• 寫在前面
• 完成的任務
• pyenv
  • 安裝pyenv
  • 安裝python 3.6.12
  • 設定當前使用者目錄使用python 3.6.12
  • 安裝pytorch 1.2
  • 安裝torchvision 0.4.0
• 寫在後面

寫在前面

2020年8月25日：這個時間點，在Jetson TX2安裝pytorch是一件非常簡單的事情，網上很多文章把它寫複雜了，實在是不應該呀

刷入JetPack 4.3，apt更換為國內源。選擇這個版本的原因是對應cuda的版本為10.0。
JetPack 4.4對應cuda版本為10.2，如果需要cuda 10.2版本，請刷入4.4。

本文基於JetPack4.3，安裝pytorch版本為1.2，torchvision版本為0.4.0

完成的任務

1. 不將pytorch安裝到系統預設的python裡
2. 安裝pyenv
3. 安裝pytorch 1.2和torchvision 0.4.0

pyenv

https://github.com/pyenv/pyenv

安裝pyenv

git clone https://github.com/pyenv/pyenv.git ~/.pyenv
echo 'export PYENV_ROOT="$HOME/.pyenv"' >> ~/.bashrc
echo 'export PATH="$PYENV_ROOT/bin:$PATH"' >> ~/.bashrc
echo -e 'if command -v pyenv 1>/dev/null 2>&1; then\n  eval "$(pyenv init -)"\nfi' >> ~/.bashrc
exec "$SHELL"
 

安裝python 3.6.12

sudo apt update; sudo apt install --no-install-recommends make build-essential libssl-dev zlib1g-dev libbz2-dev libreadline-dev libsqlite3-dev wget curl llvm libncurses5-dev xz-utils tk-dev libxml2-dev libxmlsec1-dev libffi-dev liblzma-dev

pyenv install 3.6.12

設定當前使用者目錄使用python 3.6.12

cd ~
pyenv local 3.6.12

安裝pytorch 1.2

https://forums.developer.nvidia.com/t/pytorch-for-jetson-nano-version-1-6-0-now-available/72048

下載PyTorch v1.2.0，JetPack為4.3

pip install torch-1.3.0-cp36-cp36m-linux_aarch64.whl

安裝torchvision 0.4.0

首先安裝libjpeg-dev zlib1g-dev

sudo apt install libjpeg-dev zlib1g-dev

然後安裝pillow，注意順序

pip install 'pillow<7'

最後安裝torchvision，版本一定要是0.4.0

git clone --branch v0.4.0 https://github.com/pytorch/vision torchvision 
cd torchvision
export BUILD_VERSION=0.4.0 
sudo python setup.py install 

寫在後面

大功告成，多版本python共存，pytorch不會影響到系統自帶的python