【Pythorch】win10+cuda11.1+cudnn+torch安装--新手避雷指南

0.引言

网上有关cuda安装教程很多，一步步照搬可能由于中间部件下载版本不同，导致后面无法成功安装，需要卸载重装的尴尬局面。为此，列出相关需要注意的点，请在安装前确保可下载的版本信息。

首先从获得一块显卡开始。共需要安装的部件有：

1. 显卡驱动（下载可执行文件exe安装）
2. cuda（cuda toolkits）（下载可执行文件exe安装）
3. cudnn（下载库后，复制到cuda文件夹即可）
4. torch和torchvision（pip或conda安装）

1.显卡驱动

显卡型号：NVIDIA GeForce RTX 2060
系统：win10
在NVIDIA官网下载对应版本的显卡驱动，安装比较新的就行，这个对后续影响不大。
【显卡驱动下载地址】

接下来需要找cuda和pytorch相关版本，请事先大致查一下相关的可下载版本，确保已有对应版本后在进行安装！！！

2.cuda

官网查找cuda和cudnn版本：
【cuda链接】

3.cudnn

【cudnn链接】这一步需要登录NVIDIA账号，还要填个什么问卷啥的。注意后面的版本需要与下载的cuda版本一致。

4.torch

查找可用的torch版本
【torch链接】可以查看现有torch支持的cuda版本。如图中torch现在只有cuda11.1的对应版本，所以最新出的cuda11.2还不能支持。

5.问题

<1>下载速度慢
可以设置镜像，改用国内镜像可以提速。具体可以【参考这里】

<2>报错

 File "D:\Python38\lib\site-packages\torch\__init__.py", line 129, in <module>
    raise err
OSError: [WinError 126] 找不到指定的模块。 Error loading "D:\****\torch\lib\caffe2_detectron_ops.dll" or one of its dependencies.

这个很大可能的安装版本不对应，检查安装的版本信息。
就是因为安装最新发布的cuda11.2，又有没有对应的torch，最后就一直报这个错。安装前先查询可下载版本！

<3>如何检查版本信息
cuda：win+R–>输入cmd调出命令行–>nvcc -V

cudnn：【参考这一篇】

torch：

import torch
import torchvision

print(torch.version.cuda)
print(torch.__version__)
print(torchvision.__version__)
print(torch.cuda.is_available())