Ubuntu 22 .04安装CUDA, cuDNN, TensorRT
1. 安装英伟达驱动
查看推荐的驱动版本:
ubuntu-drivers list
安装支持CUDA 12的驱动 (如:nvidia-driver-535):
sudo apt install nvidia-driver-535
重启系统。
注意:具体版本需参考 NVIDIA 官网。驱动版本 ≥525.60.13 通常兼容 CUDA 12。
2. 安装CUDA
步骤1 根据官网指导进行安装
网址:https://developer.nvidia.com/cuda-downloads
3. 安装cuDNN
步骤1 根据官网要求进行安装
网址:https://developer.nvidia.com/cudnn-downloads?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=22.04&target_type=deb_local&Configuration=Full
4. 安装TensorRT
安装网址:
https://docs.nvidia.com/deeplearning/tensorrt/latest/installing-tensorrt/installing.html
下载地址:
https://developer.nvidia.com/tensorrt/download
5 其它
5.1 驱动版本
sudo apt install nvidia-driver-535
5.2 CUDA版本
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin
sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/13.0.2/local_installers/cuda-repo-ubuntu2204-13-0-local_13.0.2-580.95.05-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2204-13-0-local_13.0.2-580.95.05-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2204-13-0-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda-toolkit-13-0
5.3 cuDNN版本
安装命令
wget https://developer.download.nvidia.com/compute/cudnn/9.14.0/local_installers/cudnn-local-repo-ubuntu2204-9.14.0_1.0-1_amd64.deb
sudo dpkg -i cudnn-local-repo-ubuntu2204-9.14.0_1.0-1_amd64.deb
sudo cp /var/cudnn-local-repo-ubuntu2204-9.14.0/cudnn-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cudnn9-cuda-12
5.4 TensorRT版本
sudo dpkg -i nv-tensorrt-local-repo-ubuntu2204-10.13.3-cuda-12.9_1.0-1_amd64.deb
sudo cp /var/nv-tensorrt-local-repo-ubuntu2204-10.13.3-cuda-12.9/nv-tensorrt-local-2F9B36B1-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get install tensorrt
5.5 Nividia官方支持矩阵
https://docs.nvidia.com/deeplearning/tensorrt/latest/getting-started/support-matrix.html