四、安装依赖库

安装pytorch和torchvision之前,先安装相关的依赖库。

1、安装pytorch依赖库



apt install build-essential make cmake cmake-curses-gui -y
apt install git g++ pkg-config curl -y
apt install libatlas-base-dev gfortran libcanberra-gtk-module libcanberra-gtk3-module -y
apt install libhdf5-serial-dev hdf5-tools -y
apt install nano locate screen -y
apt install libfreetype6-dev -y
apt install protobuf-compiler libprotobuf-dev openssl -y
apt install libssl-dev libcurl4-openssl-dev -y
apt install cython3 -y



2、安装opencv依赖库和编解码库



apt install build-essential -y
apt install libgtk2.0-dev pkg-config -y
apt install python-dev python-numpy libtbb2 libtbb-dev libjpeg-dev libpng-dev libtiff5-dev libdc1394-22-dev -y
apt install libavcodec-dev libavformat-dev libswscale-dev libv4l-dev liblapacke-dev -y
apt install libxvidcore-dev libx264-dev -y
apt install libatlas-base-dev gfortran -y
apt install ffmpeg -y



3、安装torchvision0.9.0依赖库



apt install libopenmpi2 -y
apt install libopenblas-dev -y
apt install libjpeg-dev zlib1g-dev -y



五、安装pytorch

之前我们已经知道Jetson Nano上的linux系统ubuntu18.04是类似手机的ARM架构,这也就导致它的很多包和普通的linux上的不是通用的。pytorch官网下载的包,在实际使用时无法调用开发板的显卡(这是个大问题,失去显卡的开发板算力暴跌!)。这里的pytorch以及接下来的torchvision等包都需要安装Nvidia官网给出的版本。

1、下载pytorch



#nvidia官方链接(外网,可能需要翻墙才行)

https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-9-0-now-available/72048





#如果官网无法下载,可以从百度网盘或CSDN社区下载

#BaiDu

https://pan.baidu.com/s/1Nfm9w1DfE2k_cZQ1zTykMQ

提取码:zpsa

#CSDN




2、安装pytorch

本案例中是从百度网盘下载的,里面同时包含了pytorch1.8和torchvision0.9,比较省事,将下载的压缩文件解压缩,如下图所示:




python的依赖库 pytorch依赖库_python的依赖库


打开命令窗口,进入torch目录,执行安装命令。安装过程可能需要耗费一点时间,耐心等待。


python的依赖库 pytorch依赖库_pytorch_02


3、测试pytorch

在命令窗口中依次输入以下命令,如果没有出现异常,则说明安装成功。如下图所示:


python的依赖库 pytorch依赖库_嵌入式硬件_03


六、安装torchvision

1、torchvision版本

pytorch和torchvision版本是需要对应的!可参考下表中的对应关系。


python的依赖库 pytorch依赖库_java_04


2、安装torchvision


cd torchvision # 进入到这个包的目录下

export BUILD_VERSION=0.9.0

python3 setup.py install # 安装(估计要20分钟不止吧)


安装过程如下图所示:


python的依赖库 pytorch依赖库_pytorch_05


python的依赖库 pytorch依赖库_java_06


python的依赖库 pytorch依赖库_java_07


3、测试torchvision

在命令窗口中依次输入以下命令,如果没有出现异常,则说明安装成功。如果出现如下图所示的异常,则是因为Pillow包版本导致的,在安装torchvision时自动安装了高版本的pillow,这里需要降低pillow版本,需要安装pillow==8.4.0版本。


python的依赖库 pytorch依赖库_pytorch_08


首先,卸载已经安装的pillow库。


python的依赖库 pytorch依赖库_java_09


然后,安装指定版本的pillow库。


python的依赖库 pytorch依赖库_pytorch_10


最后,再次进行测试。


python的依赖库 pytorch依赖库_java_11


七、安装JDK

1、下载JDK

前往Oracle官网下载对应版本的JDK。

https://www.oracle.com/java/technologies/downloads/


python的依赖库 pytorch依赖库_python_12


2、安装JDK

执行解压缩命令,提取到jdk-18.0.2目录,并将此目录拷贝到/usr/local/java目录下。


python的依赖库 pytorch依赖库_java_13


3、配置JDK

首先,执行命令打开bashrc文件。


gedit ~/.bashrc


然后,在文件末尾追加以下内容。


# config java env
export JAVA_HOME=/usr/local/java/jdk-18.0.2
export JRE_HOME=${JAVA_HOME}/jre
export CLASSPATH=.:${JAVA_HOME}/lib:${JRE_HOME}/lib
export PATH=.:${JAVA_HOME}/bin:$PATH


最后,重新加载bashrc环境变量信息。


source ~/.bashrc


4、测试JDK

打开命令窗口,输入以下内容,测试JDK安装是否成功。


python的依赖库 pytorch依赖库_python_14