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

by nineone, 2023-02-28

# 更新软件源
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

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

opencvjetsontensorrtcudacudnn

我爱你