Nvidia Jetson Nano Developer KIT配置全过程（三）：安装pytorch等依赖组件

四、安装依赖库

安装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 https://download.csdn.net/download/IamYZD/21027946

2、安装pytorch

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

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

3、测试pytorch

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

六、安装torchvision

1、torchvision版本

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

2、安装torchvision

cd torchvision # 进入到这个包的目录下 export BUILD_VERSION=0.9.0 python3 setup.py install # 安装（估计要20分钟不止吧）

安装过程如下图所示：

3、测试torchvision

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

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

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

最后，再次进行测试。

七、安装JDK

1、下载JDK

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

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

2、安装JDK

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

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安装是否成功。