当前位置: 代码迷 >> 综合 >> NVIDIA TX2 安裝pytorch1.8
  详细解决方案

NVIDIA TX2 安裝pytorch1.8

热度:18   发布时间:2023-12-06 07:18:05.0

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

  相关解决方案