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

润信云 技术支持

服务器ubuntu安装cuda11.0、cuDNN教程,简单易懂,包教包会

image.png

一、版本情况介绍

我的linux版本是ubuntu18.04,安装cuda11.0

提前已经安装好英伟达驱动、anaconda、gcc、make


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

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

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


安装 dkms:

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、到官网找到对应版本进行安装
image.png
image.png
image.png
image.png
image.png
image.png

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、运行完上面代码出现以下界面即代表安装成功
image.png
8、默认安装地址如下,安装按这个路径会多一个cuda-11.0的文件夹

image.png

三、设置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

image.png

验证版本

ActionScript
nvcc -V

image.png


四、cuDNN安装

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

image.png

image.png

image.png

image.png

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

image.png

本文链接:https://blog.runxinyun.com/post/227.html 转载需授权!

版权声明
网站名称: 润信云资讯网
本站提供的一切软件、教程和内容信息仅限用于学习和研究目的。
不得将上述内容用于商业或者非法用途,否则,一切后果请用户自负。
我们非常重视版权问题,如有侵权请邮件与我们联系处理。敬请谅解!邮件:7104314@qq.com
网站部分内容来源于网络,版权争议与本站无关。请在下载后的24小时内从您的设备中彻底删除上述内容。
如无特别声明本文即为原创文章仅代表个人观点,版权归《润信云资讯网》所有,欢迎转载,转载请保留原文链接。
0 85

留言0

评论

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。