玩转Jetson Nano(三):安装Pytorch GPU版
- 前言
- 安装Pytorch GPU版
- 查看CUDA版本号
- 下载Pytorch对应的wheel文件
- 测试是否安装成功
- 常见问题
- OSError: libmpi_cxx.so.20
-
- ImportError: libopenblas.so.0
-
- Illegal instruction(cpre dumped)
-
- 参考文献
前言
- 本文是个人使用Jetson Nano的电子笔记,由于水平有限,难免出现错漏,敬请批评改正。
- 更多精彩内容,可点击进入
玩转Jetson Nano专栏或我的个人主页查看
安装Pytorch GPU版
查看CUDA版本号
nvcc -V
![在这里插入图片描述](https://img-blog.csdnimg.cn/8ea36da2ec494f46a527d8fc50931165.png)
如果nvcc -v
提示未找到命令,可能是环境变量问题,则需要按如下操作:
sudo gedit ~/.bashrc
在/.bashrc中配置LD_LIBRARY_PATH路径、配置PATH路径,完整配置如下:
export LD_LIBRARY_PATH=/usr/local/cuda/lib
export PATH=$PATH:/usr/local/cuda/bin
关闭文件,并使之生效
source ~/.bashrc
即可!
下载Pytorch对应的wheel文件
https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-11-now-available/72048
![在这里插入图片描述](https://img-blog.csdnimg.cn/857fc927fbc948beb6becd8bcc8d6a7a.png)
![在这里插入图片描述](https://img-blog.csdnimg.cn/574a2baf200c4d909a59e751cf472f33.png)
![在这里插入图片描述](https://img-blog.csdnimg.cn/c746783688524652aa7da1763749655d.png)
找到torch-1.7.0-cp36-cp36m-linux_aarch64.whl所在的目录
sudo pip3 install torch-1.7.0-cp36-cp36m-linux_aarch64.whl
![在这里插入图片描述](https://img-blog.csdnimg.cn/8599f8aeb7d94feba6122716bfe6abb7.png)
![在这里插入图片描述](https://img-blog.csdnimg.cn/d71357351df2457aae610d036c940929.png)
测试是否安装成功
import torch
print(torch.__version__)
print(torch.cuda.is_available())
![在这里插入图片描述](https://img-blog.csdnimg.cn/ee0a5bee32df4f8d927ae1a8e230f4c6.png)
常见问题
OSError: libmpi_cxx.so.20
>>> import torch
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/usr/local/lib/python3.6/dist-packages/torch/__init__.py", line 189, in <module>
_load_global_deps()
File "/usr/local/lib/python3.6/dist-packages/torch/__init__.py", line 142, in _load_global_deps
ctypes.CDLL(lib_path, mode=ctypes.RTLD_GLOBAL)
File "/usr/lib/python3.6/ctypes/__init__.py", line 348, in __init__
self._handle = _dlopen(self._name, mode)
OSError: libmpi_cxx.so.20: cannot open shared object file: No such file or directory
![在这里插入图片描述](https://img-blog.csdnimg.cn/0764b013ba944ae8b60d0a5388b80d6a.png)
解决方法
进入https://pkgs.org/网站,搜索libopenmpi,下载libopenmpi2_2.1.1-8_arm64.deb
![在这里插入图片描述](https://img-blog.csdnimg.cn/35eb274eb3d44c8791419be048e109a9.png)
![在这里插入图片描述](https://img-blog.csdnimg.cn/6447b31db1ea4c3096868b60878710f5.png)
wget http://ports.ubuntu.com/pool/universe/o/openmpi/libopenmpi2_2.1.1-8_arm64.deb
![在这里插入图片描述](https://img-blog.csdnimg.cn/d39559368c714deeaccf3a343f7b43a4.png)
sudo dpkg -i libopenmpi2_2.1.1-8_arm64.deb
![在这里插入图片描述](https://img-blog.csdnimg.cn/9f3e47a7f6514e3e8cf3ba2da6c13b4c.png)
![在这里插入图片描述](https://img-blog.csdnimg.cn/85a8d8aac645421592d69231db01a5a0.png)
ImportError: libopenblas.so.0
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/usr/local/lib/python3.6/dist-packages/torch/__init__.py", line 190, in <module>
from torch._C import *
ImportError: libopenblas.so.0: cannot open shared object file: No such file or directory
![在这里插入图片描述](https://img-blog.csdnimg.cn/acbb5eb6cc154b11b01e0935561285d8.png)
解决方法
sudo apt-get install libopenblas-dev
![在这里插入图片描述](https://img-blog.csdnimg.cn/67085892ff3c46ad8488f9ea262c51bc.png)
![在这里插入图片描述](https://img-blog.csdnimg.cn/4fcb1e27cc2c49c5bac1d3ffac6c9df0.png)
Illegal instruction(cpre dumped)
>>>import torch
Illegal instruction(cpre dumped)
解决方法
具体解决方法,请点击查阅解决:Jetson系列 python3 import 报错 Illegal instruction (core dumped)
参考文献
[1] Jetson Nano Developer Kit User Guide
[2] https://docs.nvidia.com/deeplearning/frameworks/install-pytorch-jetson-platform-release-notes/index.html
[3] https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-11-now-available/72048
- 更多精彩内容,可点击进入
玩转Jetson Nano专栏或我的个人主页查看
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)