Ubuntu18.04安装CUDA及CUDNN有效方式

一、安装CUDA

首先去英伟达官网下载,下载地址:最新版,现在是CUDA Toolkit 10.2版

可以根据自己的配置要求搜不同cuda版本
https://developer.nvidia.com/cuda-toolkit-archive

下载界面如下:

这里用的Ubuntu18.04系统安装的,也可选windows。

安装命令:

wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
sudo sh cuda_10.2.89_440.33.01_linux.run

上面安装的runfile格式文件,如果vpn受限下载慢,可选择下载deb版文件,功能基本都能支持。
在这里插入图片描述
安装命令:

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 http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-10-2-local-10.2.89-440.33.01/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda

接下来默认cuda安装好了,进行配置cuda环境变量
打开终端,输入”gedit ~/.bashrc“,然后再弹出的编辑器的最后,添加如下两行

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

终端运行:source ~/.bashrc

检查:nvcc --version,如果显示下面的信息就说明安装成功了。

在这里插入图片描述

如上述方法为成功,可以试一下——同时安装驱动和cuda
这种安装方法不需要在安装cuda之前安装显卡驱动,但是安装方法比较复杂,如果前一种方法安装不成功的话可以参考这个方法。

具体步骤可以参考该链接:https://blog.csdn.net/qlulibin/article/details/78714596

二、CUDNN安装

cudnn的安装非常简单。下载地址:https://developer.nvidia.com/rdp/cudnn-archive

这里提供两种安装方式:

1.Tar File形式的安装

Tar File的下载如下图所示,选择红方框中的选项进行下载

在这里插入图片描述

下载的是cudnn-*tgz的压缩包时,按下方指令进行安装:

首先解压缩下的cudnn压缩包文件

tar -xzvf cudnn-9.0-linux-x64-v7.tgz

执行安装,其实就是拷贝头文件和库文件并给予权限

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

2.Debian File形式的安装

Debian File的下载如下图所示,选择红方框中的选项进行下载,此时可以选择下载三个
在这里插入图片描述

此时执行以下三个命令进行安装

sudo dpkg -i libcudnn7_7.0.3.11-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-dev_7.0.3.11-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-doc_7.0.3.11-1+cuda9.0_amd64.d

3.验证cudnn是否安装成功

当选择Debian File进行安装时会在/usr/src/cudnn_samples_v7有一些cudnn的例子

编译mnistCUDNN sample进行验证

cp -r /usr/src/cudnn_samples_v7/ $HOME
cd  $HOME/cudnn_samples_v7/mnistCUDNN
make clean && make
./mnistCUDNN

如果安装成功将会有如下图所示的:Test passed!
在这里插入图片描述


版权声明:本文为weixin_45591044原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。