NVIDIA Jetson nano 开发板 安装Pytorch

2022-11-14 15:45:34 浏览数 (1)

首先,我们需要更新一下软件源,关于从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

命令无回显,无报错,则为安装成功!

0 人点赞