NVIDIA Jetson nano 开发板 安装Pytorch

首先，我们需要更新一下软件源，关于从Jetson Nano开发板中更改软件源的方法，请参考：Chuanrui の 初见之旅-NVIDIA Jetson nano 开发板 Ubuntu系统更换镜像源 (1314.cool)：

更新软件源运行如下命令：

sudo apt-get -y update; 
sudo apt-get -y install autoconf bc build-essential g++-8 gcc-8 clang-8 lld-8 gettext-base gfortran-8 iputils-ping libbz2-dev libc++-dev libcgal-dev libffi-dev libfreetype6-dev libhdf5-dev libjpeg-dev liblzma-dev libncurses5-dev libncursesw5-dev libpng-dev libreadline-dev libssl-dev libsqlite3-dev libxml2-dev libxslt-dev locales moreutils openssl python-openssl rsync scons python3-pip libopenblas-dev;

运行完毕之后，我们需要安装PyTorch所依赖的其他组件：

首先，先决条件，需要安装Python3和pip3

 python3安装命令：

sudo apt-get install python3 

 pip3安装命令：

sudo apt install python3-pip

接着，我们安装Cython组件：

sudo pip3 install Cython

安装numpy：

注意，在这里，需要安装numpy的一个较低版本，否则将会提示非法指令(核心已转储)

sudo pip3 install numpy==1.19.1

安装libomp5：

如果安装后，仍然在import torch中提示libomp.so的错误，则需要卸载重装

sudo apt install libomp5 libomp-dev;

最后，安装pytorch：

从NVIDIA官网下载适用于Jeston Nano的Pytorch版本：

Pytorch 1.11.0a0 适用于Jetson Nano JetPark 4.6.1

将它放在开发板的Downloads目录下,然后使用cd命令切换至该目录：

或者直接运行以下命令：

wget "https://developer.download.nvidia.com/compute/redist/jp/v461/pytorch/torch-1.11.0a0+17540c5+nv22.01-cp36-cp36m-linux_aarch64.whl"

接下来：安装Pytorch

sudo pip3 install torch-1.11.0a0+17540c5+nv22.01-cp36-cp36m-linux_aarch64.whl

等待指令运行成功

此时，我们就已经安装好了PyTorch

运行以下命令，验证PyTorch安装：

sudo python3

import torch

命令无回显，无报错，则为安装成功！

我的博客即将同步至腾讯云开发者社区，邀请大家一同入驻：https://cloud.tencent.com/developer/support-plan?invite_code=an85r8o43fw7