网站开发公司气氛电商代运营公司排名

【linux】服务器ubuntu安装cuda11.0、cuDNN教程，简单易懂，包教包会

【创作不易，求点赞+关注+收藏】

一、版本情况介绍

我的linux版本是ubuntu18.04，安装cuda11.0
提前已经安装好英伟达驱动、anaconda、gcc、make

• 为避免安装失败，先安装以下内容

检查和安装 Linux 内核头文件：

sudo apt-get install linux-headers-$(uname -r)

安装 dkms：

sudo apt-get install dkms

安装相关依赖

sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev

二、安装cuda

1、到官网找到对应版本进行安装

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.0.1/local_installers/cuda-repo-ubuntu1804-11-0-local_11.0.1-450.36.06-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1804-11-0-local_11.0.1-450.36.06-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu1804-11-0-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda

2、对上面代码进行解释：

1、下载CUDA的pin文件，这个文件将CUDA包的优先级设为600：

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600

2、下载CUDA的本地安装包：

wget https://developer.download.nvidia.com/compute/cuda/11.0.1/local_installers/cuda-repo-ubuntu1804-11-0-local_11.0.1-450.36.06-1_amd64.deb

3、安装CUDA本地包：

sudo dpkg -i cuda-repo-ubuntu1804-11-0-local_11.0.1-450.36.06-1_amd64.deb

4、添加CUDA的GPG密钥：

sudo apt-key add /var/cuda-repo-ubuntu1804-11-0-local/7fa2af80.pub

5、更新APT包索引：

sudo apt-get update

6、安装CUDA Toolkit：

sudo apt-get -y install cuda

7、运行完上面代码出现以下界面即代表安装成功

8、默认安装地址如下，安装按这个路径会多一个cuda-11.0的文件夹

三、设置cuda环境变量

nano ~/.bashrc

添加以下行

export PATH=/usr/local/cuda-11.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

更新

source ~/.bashrc

验证版本

nvcc -V

四、cuDNN安装

1、到官网下载安装好并上传到服务器

2、进入该安装包的路径

cd /public/apps/downloads/

3、解压

tar -xvf cudnn-linux-x86_64-8.9.3.28_cuda11-archive.tar

进入解压后的目录：

cd cudnn-linux-x86_64-8.9.3.28_cuda11-archive

4、将头文件复制到 /usr/local/cuda/include/ 目录：

sudo cp include/cudnn*.h /usr/local/cuda/include/

5、将库文件复制到 /usr/local/cuda/lib64/ 目录：

sudo cp lib/libcudnn* /usr/local/cuda/lib64/

6、修改文件权限，以确保所有用户都可以读取这些文件：

sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

7、验证cuDNN安装是否成功：

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2