当前位置：服务支持 > 软件文章 > Jetson Xavier NX（eMMC）配置深度学习开发环境

Jetson Xavier NX（eMMC）配置深度学习开发环境

阅读数 3

Jetson Xavier NX (eMMC)配置深度学习开发环境

前言

本文介绍如何配置NX的深度学习开发环境。

包括以下几个部分：

(1)确定版本

(2)安装Python

(3)安装pip及换源

(4)安装Pytorch

(5)安装torchvision

1.确定版本

配置深度学习环境前要刷机成功，可参考使用SDK Manager对Jetson Xavier NX(eMMC)刷机的过程
，安装完对应Python、Pytorch的JetPack。

版本对应：使用Pytorch框架运行深度学习模型，要找好NX、JetPack、CUDA、cuDNN、Pytorch、torchvision、Python之间的版本对应关系，有一对关系不匹配就可能导致模型无法运行，所以要先确定好各自的版本。不同的Jetson设备支持不同的JetPack、同一NX可支持不同的JetPack、不同的JetPack包含不同的CUDA、cuDNN，不同的CUDA、cuDNN支持不同的Pytorch版本、不同的Pytorch版本支持不同的torchvision和Python。Jetson设备和JetPack的版本对应关系可在JetPack Archive | NVIDIA Developer中查看，JetPack和CUDA、cuDNN的版本对应关系可在JetPack Archive | NVIDIA Developer中点击相关JetPack版本查看，JetPack、Python、Pytorch的版本对应关系可在Jetson Zoo - eLinux.org、PyTorch for Jetson - Jetson & Embedded Systems / Jetson Nano - NVIDIA Developer Forums中查看，Python、Pytorch、torchvision的版本对应关系可在pytorch/vision: Datasets, Transforms and Models specific to Computer Vision (github.com)中查看。

本文的环境如下：Ubuntu 20.04，Python 3.8.10，JetPack 5.0.2，CUDA 11.4，cuDNN 8.4.1，Pytorch1.12.0，torchvision 0.13.0

2.安装Python

NX的烧录系统会带有Python，先执行命令python --version查看Python版本，Ubuntu 20.04带的是Python 3.8.10，不符合再去安装。

3.安装pip及换源

sudo apt-get install python3-pip安装，ls -a查看是否有.pip文件夹，mkdir ~/.pip没有就创建，gedit ~/.pip/pip.conf编辑，然后写入下面内容：

[global] timeout=6000 index-url=https://pypi.tuna.tsinghua.edu.cn/simple [install] trusted-host=mirrors.aliyun.com

4.安装Pytorch

通过命令下载.whl文件安装的在线方式很慢，本文通过离线下载Pytorch的.whl文件的方式安装，安装步骤可在PyTorch for Jetson - Jetson & Embedded Systems / Jetson Nano - NVIDIA Developer Forums、Installing PyTorch for Jetson Platform :: NVIDIA Deep Learning Frameworks Documentation中查看。

首先，在Jetson Zoo - eLinux.org或PyTorch for Jetson - Jetson & Embedded Systems / Jetson Nano - NVIDIA Developer Forums中下载.whl文件，一般文件会下载到Downloads文件夹。

在这里插入图片描述

然后打开Terminal，运行sudo apt-get install python3-pip libopenblas-base libopenmpi-dev libomp-dev安装依赖，然后运行pip3 install Cython，最后在下载的文件目录下，运行pip3 install <torch****.whl文件名>安装Pytorch。

5.安装torchvision

通过源码离线安装，在pytorch/vision: Datasets, Transforms and Models specific to Computer Vision (github.com)中下载.tar.gz文件。然后进入下载目录，执行命令tar -zxvf <文件名>解压文件，export BUILD_VERSION=0.x.0 # where 0.x.0 is the torchvision version，python3 setup.py install，然后等待很长时间就完成了，期间只要不报错就不需要退出，安装过程很慢。