【linux】服务器ubuntu安装cuda11.0、cuDNN教程,简单易懂,包教包会
【创作不易,求点赞+关注+收藏】
文章目录
【linux】服务器ubuntu安装cuda11.0、cuDNN教程,简单易懂,包教包会一、版本情况介绍二、安装cuda1、到官网找到对应版本进行安装2、对上面代码进行解释:1、下载CUDA的pin文件,这个文件将CUDA包的优先级设为600:2、下载CUDA的本地安装包:3、安装CUDA本地包:4、添加CUDA的GPG密钥:5、更新APT包索引:6、安装CUDA Toolkit:7、运行完上面代码出现以下界面即代表安装成功8、默认安装地址如下,安装按这个路径会多一个cuda-11.0的文件夹 三、设置cuda环境变量四、cuDNN安装1、到官网下载安装好并上传到服务器2、进入该安装包的路径3、解压进入解压后的目录:4、将头文件复制到 /usr/local/cuda/include/ 目录:5、将库文件复制到 /usr/local/cuda/lib64/ 目录:6、修改文件权限,以确保所有用户都可以读取这些文件:7、验证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.pinsudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600wget 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.debsudo dpkg -i cuda-repo-ubuntu1804-11-0-local_11.0.1-450.36.06-1_amd64.debsudo apt-key add /var/cuda-repo-ubuntu1804-11-0-local/7fa2af80.pubsudo apt-get updatesudo 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.pinsudo 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