服务器ubuntu安装cuda11.0、cuDNN教程,简单易懂,包教包会
一、版本情况介绍
我的linux版本是ubuntu18.04,安装cuda11.0
为避免安装失败,先安装以下内容
ActionScript
sudo apt-get install linux-headers-$(uname -r)
ActionScript
sudo apt-get install dkms
ActionScript
sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev
二、安装cuda
1、到官网找到对应版本进行安装






2、对上面代码进行解释:
1、下载CUDA的pin文件,这个文件将CUDA包的优先级设为600:
ActionScript
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的本地安装包:
ActionScript
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本地包:
ActionScript
sudo dpkg -i cuda-repo-ubuntu1804-11-0-local_11.0.1-450.36.06-1_amd64.deb
4、添加CUDA的GPG密钥:
ActionScript
sudo apt-key add /var/cuda-repo-ubuntu1804-11-0-local/7fa2af80.pub
5、更新APT包索引:
ActionScript
sudo apt-get update
6、安装CUDA Toolkit:
ActionScript
sudo apt-get -y install cuda
7、运行完上面代码出现以下界面即代表安装成功

8、默认安装地址如下,安装按这个路径会多一个cuda-11.0的文件夹
三、设置cuda环境变量
ActionScript
nano ~/.bashrc
ActionScript
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}}
ActionScript
source ~/.bashrc
ActionScript
nvcc -V
四、cuDNN安装
1、到官网下载安装好并上传到服务器
2、进入该安装包的路径
ActionScript
cd /public/apps/downloads/
3、解压
ActionScript
tar -xvf cudnn-linux-x86_64-8.9.3.28_cuda11-archive.tar
进入解压后的目录:
ActionScript
cd cudnn-linux-x86_64-8.9.3.28_cuda11-archive
4、将头文件复制到 /usr/local/cuda/include/ 目录:
ActionScript
sudo cp include/cudnn*.h /usr/local/cuda/include/
5、将库文件复制到 /usr/local/cuda/lib64/ 目录:
ActionScript
sudo cp lib/libcudnn* /usr/local/cuda/lib64/
6、修改文件权限,以确保所有用户都可以读取这些文件:
ActionScript
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
7、验证cuDNN安装是否成功:
ActionScript
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
本文链接:https://blog.runxinyun.com/post/227.html 转载需授权!
留言0