Nvida jetson nano 学习 : Aarch64 平台 安装GPU版本的 pytorch

	最近在学习Nvida jetson nano 一个深度学习的嵌入式平台
	记录一下安装pytorch-GPU的过程 和遇到過得錯誤

	
	首先 nvidia jetson 有自带的cuda和Python3  版本是3.6.9 
	这里要注意不要安装其他的Python 了 以免覆盖掉他的原生python

1. 修改 CUDA的路径

打开 .bashrc 文件

sudo gedit  ~/.bashrc

往文件末尾加入下面这三句 要注意 加的时候 去 /usr/local/下 看看cuda的版本免得加错了

export CUBA_HOME=/usr/local/cuda-10.2
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda-10.2/bin:$PATH

重启终端 或者 输入下面的命令 应用一下

source ~/.bashrc

2.下载 pytorch

然后我们上官方给的一个pytorch的网站下载wheel
https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-8-0-now-available/72048

我用的是 PyTorch v1.8.0这个

进入下载路径

cd 
cd Downloads

安装 命令
pip3 install torch-1.8.0-cp36-cp36m-linux_aarch64.whl

最后 在shell 上打python3 查看一下torch的版本

python3
>>> import torch
>>> torch.cuda.is_available()
True

3. import torch 有時候會報錯

OSError: libmpi_cxx.so.20: cannot open shared object file: No such file or directory

解決辦法是需要安裝一個包
dpkg -i libopenmpi2_2.1.1-8_arm64.deb

參考鏈接

第二个可能会报这个错误

>>> import torch 
Illegal instruction (core dumped)

https://blog.csdn.net/xiaosongshine/article/details/114168235