Jetson Xavier NX CUDA、cuDNN、TensorRT与Pytorch环境配置

Jetson Xavier NX CUDA、cuDNN、TensorRT与Pytorch环境配置_第1张图片

橘子大虾关注IP属地: 江苏

0.1312022.05.19 11:31:43字数 331阅读 3,854

torch与vision源码安装包下载
链接:https://pan.baidu.com/s/1mrIgGoMo0bq6otGhlh-E3A
提取码:6sb3

1. Cuda、CuDNN和TensorRT

在Jetson Xavier Nx控制台中执行指令

# 更新软件源
sudo apt update

# 安装JetPack组件包,其中包括了Cuda、CuDNN和TensorRT
sudo apt install nvidia-jetpack

# 使用vim打开~/.bashrc
vim ~/.bashrc

打开~/.bashrc 在末尾添加以下内容,将CUDA加入环境变量

export CUBA_HOME=/usr/local/cuda-10.2
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda-10.2/bin:$PATH

保存退出后,在控制台执行如下指令

# 执行以下命令使环境变量生效。
source ~/.bashrc

# 安装pip3
sudo apt install python3-pip

# 安装Jtop工具
sudo -H pip3 install -U jetson-stats

# 启动jtop
sudo jtop

启动后界面如下,可以查看到开发板资源使用情况,接着点击info

Jetson Xavier NX CUDA、cuDNN、TensorRT与Pytorch环境配置_第2张图片

image.png

如果安装成功,可以查看到CUDA、cuDNN和TensorRT版本信息

Jetson Xavier NX CUDA、cuDNN、TensorRT与Pytorch环境配置_第3张图片

image.png

2. Pytorch和Vision安装

2.1 Pytorch安装

# 更新软件源
sudo apt-get update
sudo apt-get upgrade

# 安装torch依赖
sudo apt-get install libopenblas-base libopenmpi-dev

在官网下载torch1.8版本安装包 https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-10-now-available/72048

Jetson Xavier NX CUDA、cuDNN、TensorRT与Pytorch环境配置_第4张图片

image.png

安装Cython、numpy和pytorch

pip3 install Cython
pip3 install numpy torch-1.8.0-cp36-cp36m-linux_aarch64.whl

安装成功后执行python3,输入红框中代码执行,验证torch是否安装成功,如果返回值为True则表示安装成功。

Jetson Xavier NX CUDA、cuDNN、TensorRT与Pytorch环境配置_第5张图片

image.png

2.2 Vision安装

下载 vision-0.9.0

git clone --branch  v0.9.0 https://github.com/pytorch/vision torchvision

解压vision-0.9.0,进入vision-0.9.0文件夹,打开控制台

image.png

执行以下命令编译安装torchvision(时间比较长)

export BUILD_VERSION=0.9.0
sudo python3 setup.py install –user

安装成功如下图

Jetson Xavier NX CUDA、cuDNN、TensorRT与Pytorch环境配置_第6张图片

image.png

升级pillow

pip3 install –upgrade pillow -I [https://pypi.tuna.tsinghua.edu.cn/simple](https://pypi.tuna.tsinghua.edu.cn/simple)

切换到非vision-0.9.0目录,尝试导入python库

image.png

成功导入,则安装成功。

你可能感兴趣的:(pytorch,人工智能,python)