首先,我们需要更新一下软件源,关于从Jetson Nano开发板中更改软件源的方法,请参考:Chuanrui の 初见之旅-NVIDIA Jetson nano 开发板 Ubuntu系统更换镜像源 (1314.cool):
更新软件源运行如下命令:
代码语言:javascript复制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安装命令:
代码语言:javascript复制sudo apt install python3-pip
接着,我们安装Cython组件:
代码语言:javascript复制sudo pip3 install Cython
安装numpy:
注意,在这里,需要安装numpy的一个较低版本,否则将会提示非法指令(核心已转储)
代码语言:javascript复制sudo pip3 install numpy==1.19.1
安装libomp5:
如果安装后,仍然在import torch中提示libomp.so的错误,则需要卸载重装
代码语言:javascript复制sudo apt install libomp5 libomp-dev;
最后,安装pytorch:
从NVIDIA官网下载适用于Jeston Nano的Pytorch版本:
Pytorch 1.11.0a0 适用于Jetson Nano JetPark 4.6.1
将它放在开发板的Downloads目录下,然后使用cd命令切换至该目录:
或者直接运行以下命令:
代码语言:javascript复制wget "https://developer.download.nvidia.com/compute/redist/jp/v461/pytorch/torch-1.11.0a0 17540c5 nv22.01-cp36-cp36m-linux_aarch64.whl"
接下来:安装Pytorch
代码语言:javascript复制sudo pip3 install torch-1.11.0a0 17540c5 nv22.01-cp36-cp36m-linux_aarch64.whl
等待指令运行成功
此时,我们就已经安装好了PyTorch
运行以下命令,验证PyTorch安装:
代码语言:javascript复制sudo python3
代码语言:javascript复制import torch
命令无回显,无报错,则为安装成功!