CUDA11.4、CUDNN、Pytorch安装

显卡驱动查看

  1. 桌面任意位置右击,选择 NVIDIA控制面板
  2. 点击3D设置中 通过预览调整图像设置
  3. 点击左下方的 系统信息
  4. 再点击 组件
    如下图所示:
    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等,即可搜索到相应的版本,方便简单。如下图所示:

CUDA11.4、CUDNN、Pytorch安装

; CUDNA安装过程

首先以管理员身份运行CUDA11.4:

CUDA11.4、CUDNN、Pytorch安装

CUDA环境配置

  1. 我的电脑–>右击属性–>高级系统设置
  2. 点击系统变量中的 path.

CUDA11.4、CUDNN、Pytorch安装
3. 添加如下路径即可(参考博文)这里可以多参考几篇博客,有的人添加的路径不太一样。

CUDA11.4、CUDNN、Pytorch安装
再验证CUDA是否安装正确:
cmd进入命令窗口:输入: nvcc -V:如下图所示:
CUDA11.4、CUDNN、Pytorch安装
可以看到,电脑的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安装

; 检查cnDNN是否安装成功:

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

  1. 进入到cuda的安装路径,C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\demo_suite,找到如下两个.exe文件:
    CUDA11.4、CUDNN、Pytorch安装
  2. 首先执行: deviceQuery.exe,查看是否出现如下界面:
    CUDA11.4、CUDNN、Pytorch安装
    CUDA11.4、CUDNN、Pytorch安装

; 3.安装Pytorch

Pytorch官网
1:进入Pytorch官网
注意:CUDA11.4可以安装CUDA11.3版本对应的pytorch。

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

Original: https://blog.csdn.net/weixin_54546190/article/details/122097856
Author: ☞源仔
Title: CUDA11.4、CUDNN、Pytorch安装

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/625838/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球