文章目录
写在前面
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
https://github.com/pyenv/pyenv
安装pyenv
git clone https://github.com/pyenv/pyenv.git ~/.pyenv
echo 'export PYENV_ROOT="$HOME/.pyenv"'