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