Jetson开发板配置pytorch环境指南_jetson pytorch190

1. 切换python默认版本

1.1 方法1：

1. 看一下可用python版本：

sudo update-alternatives --config python
1

应出现如下界面：
若未出现，则说明未建立软链接，不要紧，用方法2

• 0 /usr/bin/python2.7 2 自动模式
  1 /usr/bin/python2.7 2 手动模式
  2 /usr/bin/python3.6 1 手动模式
  要维持当前值[*]请按回车键，或者键入选择的编号：

2. 通过键入2+回车，使默认python解释器更改为python3.6

1.2 方法2：

1. 直接指定Python3.6解释器为最高优先级

sudo update-alternatives --install /usr/bin/python python /usr/bin/python3.6 1
1

2. 安装pip

下载get-pip.py后直接运行即可。

• 可以到我的网盘下：
  提取码：7dhg
• 注意：pip会安装在你运行‘get-pip.py’命令的python解释器环境下

3. torch及torchvision安装

可以看一下官方教程：Nvidia Nx 官方安装torch步骤

• torch安装
  按句执行，注意第一行是官方下载torch1.x.whl的命令，一般网速很慢，
  可以到网盘下载：
  注意第4句要在你安装torch.whl的文件路径下执行

wget https://nvidia.box.com/shared/static/p57jwntv436lfrd78inwl7iml6p13fzh.whl -O torch-1.6.0-cp36-cp36m-linux_aarch64.whl
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev 
pip3 install Cython
pip3 install torch-1.6.0-cp36-cp36m-linux_aarch64.whl
1
2
3
4

• torchvision安装
  安装与torch版本匹配的torchvision，我这里装的是torch1.6+torchvision0.7.0,
  你需要根据你torch的版本改变第2、4行的torchvision版本号
  torch与torchvision版本对应关系见官网或下一节

sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
git clone --branch v0.7.0 https://gitee.com/zero-one-game/vision torchvision  # see below for version of torchvision to download
cd torchvision
export BUILD_VERSION=0.7.0  # where 0.x.0 is the torchvision version  
python3 setup.py install --user
cd ../  # attempting to load torchvision from build dir will result in import error
1
2
3
4
5
6

• torch与torchvision版本对应关系：
  

4. 安装其他常用库

• 一般会用到opencv-python，numpy, pandas, matplotlib等库，建议从默认地址下载，即：
  例：

pip install opencv-python
1

• 如果速度慢，可以临时换源：
  例：

pip install opencv-python -i http://pypi.mirrors.ustc.edu.cn/simple/ 
sudo apt-get install python3-matplotlib

1
2
3

• 忘了simple后要不要加，试一下吧
• 国内源地址：

阿里云 http://mirrors.aliyun.com/pypi/simple/
豆瓣(douban) http://pypi.douban.com/simple/
清华大学 https://pypi.tuna.tsinghua.edu.cn/simple/
中国科学技术大学 http://pypi.mirrors.ustc.edu.cn/simple/

5. 可能遇到的麻烦

5.1 使用Python时的出现“Illegal instruction(cpre dumped)”

• 解决方法：运行python前先执行命令OPENBLAS_CORETYPE=ARMV8
  例：

OPENBLAS_CORETYPE=ARMV8
python ***.py
1
2

• 长期解决方法：
  将export OPENBLAS_CORETYPE=ARMV8加入到~/.bashrc

5.2 使用pip时出现ModuleNotFoundError: No module named ‘pip’

• 解决方法：用你的python再执行一次

python get-pip.py
1

5.3 torch安装出错

• 问题描述：执行第4节以下命令后报错：

pip3 install numpy torch-1.6.0-cp36-cp36m-linux_aarch64.whl
1

• 问题原因： torch.whl文件不完整
• 解决方法：
  重新到网上找一个torch.whl资源下载。