admin 管理员组

文章数量: 887021

目录

    • 0.引言
    • 1.显卡驱动
    • 2.cuda
    • 3.cudnn
    • 4.torch
    • 5.问题

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())

本文标签： 新手 指南 Pythorch Torch CUDNN

版权声明：本文标题：【Pythorch】win10+cuda11.1+cudnn+torch安装--新手避雷指南 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://www.freenas.com.cn/jishu/1716185758h669945.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。