python多版本共存 ubuntu_Ubuntu18.04-多个Python版本共存-pyenv-Jetson TX2-JetPack 4.3-pytorch安装-torchvision...

写在前面

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

完成的任务

不将pytorch安装到系统默认的python里

安装pyenv

安装pytorch 1.2和torchvision 0.4.0

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

下载PyTorch v1.2.0，JetPack为4.3

pip install torch-1.2.0a0+8554416-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