ubuntu安装cuda,cudnn,tensorrt步骤,以及注意事项

cuda,cudnn,tensorrt安装包网址

下载cuda安装包网址:https://developer.nvidia.com/cuda-toolkit-archive
下载cudnn安装包网址: https://developer.nvidia.com/rdp/cudnn-archive
TensorRT官网:https://developer.nvidia.com/tensorrt
TensorRT下载:https://developer.nvidia.com/nvidia-tensorrt-download

选择与显卡匹配的cuda

nvidia-smi                # 查看 CUDA 版本和显卡驱动型号
cat /proc/version         # 查看 Ubuntu / CentOS 版本
uname -a                  # 查看 linux 的内核版本和系统是多少位

CUDA11.3安装

1. cuda安装

(1) cuda采用deb安装命令

deb(local)

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
wget https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda-repo-ubuntu2004-11-3-local_11.3.0-465.19.01-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-3-local_11.3.0-465.19.01-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-3-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda

deb(network)

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences<

猜你喜欢

转载自blog.csdn.net/m0_47405013/article/details/129465341