CUDA11.4、CUDNN、Pytorch安装

显卡驱动查看

  1. 桌面任意位置右击,选择NVIDIA控制面板
  2. 点击3D设置中通过预览调整图像设置
  3. 点击左下方的系统信息
  4. 再点击组件
    如下图所示:
    CUDA11.4、CUDNN、Pytorch安装_第1张图片

安装CUDA

下载CUDA

百度盘

本人已经下载好的CUDA1.4.1和对应版本的CUDNN:
链接:https://pan.baidu.com/s/1Vixfl7tv4I6-OuBMF2ITQw
提取码:b2b7
–来自百度网盘超级会员V3的分享

自行下载

**法一:**可进入CUDA Toolkit 官网进行下载,但是一般都是最新的CUDA版本,可能不适合你的显卡版本,但其他版本在官网不太好找。
**法二:**直接百度输入CUDA TOOlkit 11.4或10.0等,即可搜索到相应的版本,方便简单。如下图所示:
CUDA11.4、CUDNN、Pytorch安装_第2张图片CUDA11.4、CUDNN、Pytorch安装_第3张图片

CUDNA安装过程

首先以管理员身份运行CUDA11.4:
CUDA11.4、CUDNN、Pytorch安装_第4张图片CUDA11.4、CUDNN、Pytorch安装_第5张图片CUDA11.4、CUDNN、Pytorch安装_第6张图片CUDA11.4、CUDNN、Pytorch安装_第7张图片CUDA11.4、CUDNN、Pytorch安装_第8张图片CUDA11.4、CUDNN、Pytorch安装_第9张图片CUDA11.4、CUDNN、Pytorch安装_第10张图片CUDA11.4、CUDNN、Pytorch安装_第11张图片CUDA11.4、CUDNN、Pytorch安装_第12张图片CUDA11.4、CUDNN、Pytorch安装_第13张图片

CUDA环境配置

  1. 我的电脑–>右击属性–>高级系统设置
  2. CUDA11.4、CUDNN、Pytorch安装_第14张图片
  3. 点击系统变量中的path.
    CUDA11.4、CUDNN、Pytorch安装_第15张图片
  4. 添加如下路径即可(参考博文)这里可以多参考几篇博客,有的人添加的路径不太一样。

CUDA11.4、CUDNN、Pytorch安装_第16张图片
再验证CUDA是否安装正确:
cmd进入命令窗口:输入:nvcc -V:如下图所示:
CUDA11.4、CUDNN、Pytorch安装_第17张图片
可以看到,电脑的CUDA 11.4安装成功。

cuDNN下载及安装

cuDNN下载官网
cuDNN是用于深度神经网络的GPU加速库,在具体编写神经网络程序时会使用到。CUDA与cuDNN的关系可参见这篇文章。

与安装CUDA需选择对应的版本类似,安装cuDNN时也需选择与CUDA对应的版本。

本文给出目前已知CUDA与cuDNN的版本对应关系,具体的安装包及支持情况可查询cuDNN官网。需要特别说明的一点是,cuDNN的下载需登录NVIDIA账户,如若没有可注册一个方便后续安装(注册很快,回答几个邮箱问题即可)。
安装步骤:

  1. 把下载好的cuDNN的zip文件解压。
  2. 再把解压后的三个文件夹复制到CUDA安装的目录中即可。
    CUDA11.4、CUDNN、Pytorch安装_第18张图片

检查cnDNN是否安装成功:

检查cnDNN是否安装成功:博客

  1. 进入到cuda的安装路径,C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\demo_suite,找到如下两个.exe文件:
    CUDA11.4、CUDNN、Pytorch安装_第19张图片
  2. 首先执行:deviceQuery.exe,查看是否出现如下界面:
    CUDA11.4、CUDNN、Pytorch安装_第20张图片
  3. 然后执行bandwidthTest.exe,出现如下界面,则代代表cuDNN也安装成功。CUDA11.4、CUDNN、Pytorch安装_第21张图片

3.安装Pytorch

Pytorch官网
1:进入Pytorch官网
注意:CUDA11.4可以安装CUDA11.3版本对应的pytorch。
CUDA11.4、CUDNN、Pytorch安装_第22张图片

创建虚拟环境

创建虚拟环境

conda create -n evn-Name python==xx.xx

激活虚拟环境

activate env-Name

安装Pytorch

因为Pytorch较大,安装时间较长,耐心等待。无需用镜像源,也就10分钟左右。

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

检测pytorch是否安装成功就很简单了,输入python,再import torch,正常导入即成功安装。

参考博客

windows10安装cuda11.4
Cuda和cuDNN安装教程(超级详细)
Windows10检查Cuda和cuDNN是否安装成功?
Win10安装CUDA11.4和Torch1.9.0

你可能感兴趣的:(python,深度学习,pytorch,深度学习,python)