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

完成的任务

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

参与评论 您还未登录,请先 登录 后发表或查看评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
©️2022 CSDN 皮肤主题:大白 设计师:CSDN官方博客 返回首页

打赏作者

zhouwensmile

你的鼓励将是我创作的最大动力

¥2 ¥4 ¥6 ¥10 ¥20
输入1-500的整数
余额支付 (余额:-- )
扫码支付
扫码支付:¥2
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值