【linux】服务器ubuntu安装cuda11.0、cuDNN教程,简单易懂,包教包会
【创作不易,求点赞+关注+收藏】
文章目录
- 【linux】服务器ubuntu安装cuda11.0、cuDNN教程,简单易懂,包教包会
- 一、版本情况介绍
- 二、安装cuda
- 1、到官网找到对应版本进行安装
- 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
二、安装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