安装CUDA10和CUDNN760

  • 2019 年 11 月 28 日
  • 筆記

安装CUDA10

1.去官网下载想要的cuda

https://developer.nvidia.com/cuda-toolkit-archive

2.选中一个 下载包,

按照自己电脑的情况选相应的选项,注意最后一行我选的是deb(network)方式安装,因为如果用deb(local)的话,你可以试着选那个,然后看下面的内容,需要下载整个包,有1.7G,还是算了。这个不影响,就相当于你下载英雄联盟是选直接下载安装包,还是现下载腾讯游戏安装器这种感觉把

这里我选择 network, 按照如下的提示来就行了:

现download那个2.8KB的东西,在下图右上角

然后找到那个东西的下载路径,执行如上图提示的语句:

sudo dpkg -i cuda-repo-ubuntu1604_10.0.130-1_amd64.deb

sudo apt-key adv –fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub

此时可能会出现一个错误:

gpgkeys: protocol `https' not supported

解决方案:

sudo apt install gnupg-curl

sudo apt-get update

sudo apt-get install cuda

然后等待其下载就好

然后成功安装,重启电脑使其生效。

(注:如果你觉得通过这种方式下载较慢,那可以选择先下载好那个1.7G的本地包那个方式安装,依照各自情况而定,不过一般来说这个会快一些,我用了7分左右就下完了)

3.验证我刚刚CUDA是否安装成功

可以在 /usr/local/cuda 路径下面找到一个samples文件夹:

进入此文件夹

cd /usr/local/cuda/samples

sudo make all -j8

完成之后,找到device_query命令,我是在 samples/1_Utilities/deviceQuery目录 下找到的,如图所示:

然后执行 ./deviceQuery

会显示当前电脑上的GPU设备的详细信息,我这是用我的笔记本电脑在作演示,所以GPU就很粗糙,

是GeForce GTX-950M 只有4046MB, 打游戏还是够的,

若成功显示以上信息,至此CUDA安装成功

安装CUDNN

1.下载cudnn7.6.0 安装包

https://developer.nvidia.com/cudnn

然后可能需要注册一个帐号才能继续,

选择:

然后选择自己要的版本,本文以CUDNN7.6.0为例

默认显示的是4个最新版,我选最下面的历史版本:Archived cuDNN Releases

然后选择:

随后弹出:

可以选择下载对应的Ubantu 的三个deb包:Runtime运行环境,Developer开发库,CodeSamples示例,然后安装。

官网也有安装说明:https://docs.nvidia.com/deeplearning/sdk/cudnn-install/#installlinux-deb

我选择直接用Linux通用版本,cuDNN Libraryfor Linux,反正照样能用

官网也有说明:https://docs.nvidia.com/deeplearning/sdk/cudnn-install/#installlinux-tar

2.本地操作

下载下来的文件为:

cudnn-10.0-linux-x64-v7.6.0.64.solitairetheme8

改个名后解压缩:

mv cudnn-10.0-linux-x64-v7.6.0.64.solitairetheme8 cudnn-10.0-linux-x64-v7.6.0.64.tgz

tar -zxvf cudnn-10.0-linux-x64-v7.6.0.64.tgz

得到一个文件夹:

里面的内容如下:

进入得到的cuda文件夹,复制一些东西到我们之前安装cuda的路径下(/usr/local/cuda/)

cd cuda

sudo cp lib64/lib* /usr/local/cuda/lib64/

sudo cp include/cudnn.h /usr/local/cuda/include/

接下来 进入安装cuda的目录:

cd /usr/local/cuda/lib64

更新cuDNN库文件的软连接,命令如下:

sudo chmod +r libcudnn.so.7.6.0

sudo ln -sf libcudnn.so.7.6.0 libcudnn.so.7

sudo ln -sf libcudnn.so.7 libcudnn.so

sudo ldconfig

注释:因为我们安装的cudnn是7.6.0,所以上面的文件后面的数字是7.6.0 或者 7

如果安装的是其他版本的cudnn,那么相应的数字会依据版本号而作修改

至此,CUDNN安装完毕