下载和安装CUDA和Cudnn(图文详解)

Pytorch环境需要用到CUDA,所以我们要安装CUDA的驱动。这里我安装的是CUDA10.0,对应的Cudnn是7.4.1。

一、下载

CUDA10.0的官网地址,选择好自己所需版本,这里是我推荐的选择,然后点击Download。
下载和安装CUDA和Cudnn(图文详解)_第1张图片
cudnn官网地址,直接寻找并下载7.4.1.5版本的cudnn。
下载和安装CUDA和Cudnn(图文详解)_第2张图片
下载完后应该是这两个文件。
在这里插入图片描述

二、安装

1、打开cuda.exe进行安装,可能会出现下面的提示,选择OK即可。

下载和安装CUDA和Cudnn(图文详解)_第3张图片
下载和安装CUDA和Cudnn(图文详解)_第4张图片
2、同意并继续。
下载和安装CUDA和Cudnn(图文详解)_第5张图片
3、这里我们选择【自定义】,然后下一步。
下载和安装CUDA和Cudnn(图文详解)_第6张图片
4、这里是安装的驱动程序组件,默认全选,然后下一步。
下载和安装CUDA和Cudnn(图文详解)_第7张图片
5、直接默认装在C盘,这样可以防止各种出错。
下载和安装CUDA和Cudnn(图文详解)_第8张图片
6、这里提示没有CUDA的编辑器,不用管它,勾选了然后Next,等待安装。
下载和安装CUDA和Cudnn(图文详解)_第9张图片
7、可能有些同学会报安装失败,这里不用担心,只是一两个工具没有装上,不影响。安装完成后,解压刚刚下载的第二个文件,即Cudnn的压缩包,然后进入cuda目录,复制这些文件。
下载和安装CUDA和Cudnn(图文详解)_第10张图片
8、进入C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0中,将刚刚复制的文件粘贴到这里,到此安装完成。
下载和安装CUDA和Cudnn(图文详解)_第11张图片
9、最终安装成功的目录如下,大家可以对照一下:
下载和安装CUDA和Cudnn(图文详解)_第12张图片

你可能感兴趣的:(深度学习,cuda,cudnn,nvidia,pytorch,anaconda)