当前位置:   article > 正文

Win10 安装 CUDA(CUDA Toolkit)、cuDNN + PyTorch 详细教程_cuda安装

cuda安装

Win10 PyTorch 安装 详细教程(以 pytorch)

1. 查看驱动版本,确定能安装的 CUDA Toolkit 版本

Win+r 打开 cmd 运行 nvidia-smi 命令

在这里插入图片描述

我的 CUDA Version 是 11.2 ,那么就可以下载 ≤11.2 的CUDA Toolkit。因为 CUDA 11.1 直接适应的 PyTorch 版本比较多,所以我选了 CUDA 11.1 。

如果命令运行失败,右击Windows图标打开 设备管理器 看看自己的电脑是否有支持 GPU 的显卡。

在这里插入图片描述

2. 下载并安装 CUDA Toolkit

  1. 下载
    点开链接 CUDA Toolkit官方下载地址 找到 CUDA 11.0 直接下载即可:

    在这里插入图片描述

    其他版本的自行找到对应版本下载就成。

  2. 安装:

    注意:在这里插入图片描述

3. 下载cuDNN,解压后放到 CUDA Toolkit 的目录下

  1. 下载
    点开链接 cuDNN官方下载地址 ,找到 对应 CUDA Toolkit 版本的 cuDNN 下载即可。

    在这里插入图片描述

    找到对应 CUDA 11.1 的 cuDNN 点击下载,我还专门找了个和我的 CUDA Toolkit 发布时间一样的。声明:本文内容由网友自发贡献,转载请注明出处:【wpsshop博客】

推荐阅读
相关标签