1、下载TX2专用torch包(.whl)不能去pytorch的官网找安装指令。
TX2专用下载地址,要和jetpack的版本对应好(网址里也有很清晰的安装教程)pytorch.whl下載
2、配备环境以及安装
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev
pip install Cython
pip install numpy torch-1.8.0-cp36-cp36m-linux_aarch64.whl
安装慢可添加国内镜像源
清华:https://pypi.tuna.tsinghua.edu.cn/simple
阿里云:http://mirrors.aliyun.com/pypi/simple/
中国科技大学:https://pypi.mirrors.ustc.edu.cn/simple/
华中理工大学:http://pypi.hustunique.com/
山东理工大学:http://pypi.sdutlinux.org/
豆瓣:http://pypi.douban.com/simple/
注意:python版本必须和pytorch版本对应,否则安装不上;python3.6对应的是pytorch1.8
检测pytorch是否安装成功:
python
>>>import torch
>>>torch.cuda.is_available()
>>>True
若出现:
>>> import torch
Illegal instruction (core dumped)
问题解决
在网上寻找解决方案时,看到了这个网页:https://stackoverflow.com/questions/65631801/illegal-instructioncore-dumped-error-on-jetson-nano
解决的方法就是增加:OPENBLAS_CORETYPE=ARMV8
可以使用临时添加方法,在运行Python指令前运行:
export OPENBLAS_CORETYPE=ARMV8
也可以采用增加系统变量方法,可以进行全局修改。
将“export OPENBLAS_CORETYPE=ARMV8
”加入到“~/.bashrc”中
想弄清楚原因,可以查阅这个:
https://www.reddit.com/r/JetsonNano/comments/ktcyoh/illegal_instructioncore_dumped_error_on_jetson/
有时候会报错
OSError: libmpi_cxx.so.20: cannot open shared object file: No such file or directory
解決辦法是需要安裝一個包
dpkg -i libopenmpi2_2.1.1-8_arm64.deb
参考链接
ImportError: libopenblas.so.0: cannot open shared object file: No such file or directory
3、安装 torchvision
sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
pip3 install setuptools #这个官方教程没写
git clone --branch v0.9.0 https://github.com/pytorch/vision torchvision
cd torchvisionsudo python3 setup.py install
sudo python3 setup.py install若失败,去掉sudo再执行
4、测试
python3
>>>import torch
>>>print(torch.cuda.is_available())
出现true一般就可以了,需要其他的库就使用pip3