ubuntu 20.04 環(huán)境下安裝CUDA 11.8, cuDNN v8.6.0和TensorRT 8.6.0（deb方式）

1 安裝

1-1 安裝cuda 11.8（deb方式）

sudo wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
sudo wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda-repo-ubuntu2004-11-8-local_11.8.0-520.61.05-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-8-local_11.8.0-520.61.05-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2004-11-8-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda

取消勾選驅(qū)動, 下一步

添加環(huán)境變量

sudo vim ~/.bashrc

最后一行添加：

export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:/usr/local/cuda-11.8/extras/CPUTI/lib64
export CUDA_HOME=/usr/local/cuda-11.8/bin
export PATH=$PATH:$LD_LIBRARY_PATH:$CUDA_HOME

保存關(guān)閉后，應(yīng)用一下更改

source ~/.bashrc

1-2 安裝cudnn v8.6.0（deb方式）

下載 cuDNN v8.6.0 for CUDA 11.x

sudo dpkg -i cudnn-local-repo-ubuntu2004-8.6.0.163_1.0-1_amd64.deb 

導(dǎo)入CUDA GPG key

sudo cp /var/cudnn-local-repo-ubuntu2004-8.6.0.163/cudnn-local-B0FE0A41-keyring.gpg /usr/share/keyrings/

刷新存儲庫元數(shù)據(jù)

sudo apt-get update

進入 cudnn-local 目錄

cd /var/cudnn-local-repo-ubuntu2004-8.6.0.163

安裝

sudo apt-get install libcudnn8=8.6.0.163-1+cuda11.8
sudo apt-get install libcudnn8-dev=8.6.0.163-1+cuda11.8
sudo apt-get install libcudnn8-samples=8.6.0.163-1+cuda11.8

1-3 安裝 TensorRT （deb方式）

sudo dpkg -i nv-tensorrt-local-repo-ubuntu2004-8.6.0-cuda-11.8_1.0-1_amd64.deb 
sudo cp /var/nv-tensorrt-local-repo-ubuntu2004-8.6.0-cuda-11.8/*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
# 安裝 tensorrt
sudo apt-get install tensorrt
# 如果使用 Python
sudo apt-get install python3-libnvinfer-dev
# 如果轉(zhuǎn)換 onnx 模型
sudo apt-get install onnx-graphsurgeon
# 如果轉(zhuǎn)換 TensorFlow 模型
sudo apt-get install uff-converter-tf
# 驗證 TensorRT 是否安裝成功
dpkg-query -W tensorrt
# tensorrt        8.6.0.12-1+cuda11.8

2 卸載

2-1 卸載 cudnn

sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn*
# 查詢
sudo dpkg -l | grep cudnn
# 將查詢結(jié)果全部卸載
sudo dpkg -r libcudnn8-samples
sudo dpkg -r libcudnn8-dev
sudo dpkg -r libcudnn8
sudo dpkg -r cudnn-local-repo-ubuntu2004-8.6.0.163

2-2 卸載 cuda

sudo apt-get remove cuda
sudo apt autoremove 
sudo apt-get remove cuda*
 
#刪除對應(yīng)的cuda版本文件夾
cd /usr/local/
sudo rm -r cuda-****
 
#查看剩余的殘留文件
sudo dpkg -l |grep cuda
#卸載對應(yīng)的殘留文件
sudo dpkg -P 殘留文件名

參考文檔：
1 NVIDIA CUDA Installation Guide for Linux
2 NVIDIA cuDNN Documentation
3 NVIDIA Deep Learning TensorRT Documentation文章來源地址http://www.zghlxwxcb.cn/news/detail-494141.html

到了這里，關(guān)于ubuntu 20.04 環(huán)境下安裝CUDA 11.8, cuDNN v8.6.0和TensorRT 8.6.0（deb方式）的文章就介紹完了。如果您還想了解更多內(nèi)容，請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章，希望大家以后多多支持TOY模板網(wǎng)！