python|CUDA11.4、CUDNN、Pytorch安装

显卡驱动查看

  1. 桌面任意位置右击,选择NVIDIA控制面板
  2. 点击3D设置中通过预览调整图像设置
  3. 点击左下方的系统信息
  4. 再点击组件
    如下图所示:
    python|CUDA11.4、CUDNN、Pytorch安装
    文章图片
安装CUDA 下载CUDA 百度盘
本人已经下载好的CUDA1.4.1和对应版本的CUDNN:
链接:https://pan.baidu.com/s/1Vixfl7tv4I6-OuBMF2ITQw
提取码:b2b7
–来自百度网盘超级会员V3的分享
自行下载
**法一:**可进入CUDA Toolkit 官网进行下载,但是一般都是最新的CUDA版本,可能不适合你的显卡版本,但其他版本在官网不太好找。
**法二:**直接百度输入CUDA TOOlkit 11.4或10.0等,即可搜索到相应的版本,方便简单。如下图所示:
python|CUDA11.4、CUDNN、Pytorch安装
文章图片
python|CUDA11.4、CUDNN、Pytorch安装
文章图片

CUDNA安装过程 首先以管理员身份运行CUDA11.4:
python|CUDA11.4、CUDNN、Pytorch安装
文章图片
python|CUDA11.4、CUDNN、Pytorch安装
文章图片
python|CUDA11.4、CUDNN、Pytorch安装
文章图片
python|CUDA11.4、CUDNN、Pytorch安装
文章图片
python|CUDA11.4、CUDNN、Pytorch安装
文章图片
python|CUDA11.4、CUDNN、Pytorch安装
文章图片
python|CUDA11.4、CUDNN、Pytorch安装
文章图片
python|CUDA11.4、CUDNN、Pytorch安装
文章图片
python|CUDA11.4、CUDNN、Pytorch安装
文章图片
python|CUDA11.4、CUDNN、Pytorch安装
文章图片

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

再验证CUDA是否安装正确:
cmd进入命令窗口:输入:nvcc -V:如下图所示:
python|CUDA11.4、CUDNN、Pytorch安装
文章图片

可以看到,电脑的CUDA 11.4安装成功。
cuDNN下载及安装 cuDNN下载官网
cuDNN是用于深度神经网络的GPU加速库,在具体编写神经网络程序时会使用到。CUDA与cuDNN的关系可参见这篇文章。
【python|CUDA11.4、CUDNN、Pytorch安装】与安装CUDA需选择对应的版本类似,安装cuDNN时也需选择与CUDA对应的版本。
本文给出目前已知CUDA与cuDNN的版本对应关系,具体的安装包及支持情况可查询cuDNN官网。需要特别说明的一点是,cuDNN的下载需登录NVIDIA账户,如若没有可注册一个方便后续安装(注册很快,回答几个邮箱问题即可)。
安装步骤:
  1. 把下载好的cuDNN的zip文件解压。
  2. 再把解压后的三个文件夹复制到CUDA安装的目录中即可。
    python|CUDA11.4、CUDNN、Pytorch安装
    文章图片
检查cnDNN是否安装成功: 检查cnDNN是否安装成功:博客
  1. 进入到cuda的安装路径,C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\demo_suite,找到如下两个.exe文件:
    python|CUDA11.4、CUDNN、Pytorch安装
    文章图片
  2. 首先执行:deviceQuery.exe,查看是否出现如下界面:
    python|CUDA11.4、CUDNN、Pytorch安装
    文章图片
  3. 然后执行bandwidthTest.exe,出现如下界面,则代代表cuDNN也安装成功。python|CUDA11.4、CUDNN、Pytorch安装
    文章图片
3.安装Pytorch Pytorch官网
1:进入Pytorch官网
注意:CUDA11.4可以安装CUDA11.3版本对应的pytorch。
python|CUDA11.4、CUDNN、Pytorch安装
文章图片

创建虚拟环境 创建虚拟环境
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

    推荐阅读