CUDA,cuDNN,pytorch安装(win)

阿里云2000元红包!本站用户参与享受九折优惠!

本来不怎么想搞的,但是想一想百度到的都是一年前的内容,而且安装还比较复杂,所以决定还是写一下吧。当然现阶段我还不知道我配置的这个有什么问题。

import torch报错
ImportError:DLL load failed :系统找不到指定的模块。
解决办法(GPU版本的torch):装CUDA,装cuDNN,再重新装torch


CUDA安装

这个和NVIDIA的驱动不一样,更新驱动的时候只安装了一部分程序。下载地址


选择平台


选择下载

下载完成之后直接安装,下面的√我是全部选上的(头铁)


开始界面


头铁

接下来就是为CUDA配置环境变量了,我就不和很多教程一样配置那么多环境变量了,找到CUDA的安装路径,默认是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1 。根据你的实际情况找找看。下面是在用户环境变量中配置的四个环境变量:

CUDA环境变量

看到这里应该是明白安装的时候需要装什么了吧,其实就需要一个GPU Computing,哈哈哈。
CUDA 还需要添加cuDNN,虽然不知道有啥用,不过并不算复杂。

cuDNN安装

下载地址
这个需要登录,填写问卷调查之后才能下载


cuDNN下载页面


文件结构

只需要将里面的文件对应拷贝到CUDA安装目录下对应的位置即可。
这样子CUDA就全部安装完成了,接下来就可以安装pytorch的GPU版本了。如果上面的步骤没有做就会遇到torch导入失败的情况,虽然torch安装的时候也是一切顺利。


torch导入失败

装完上面的CUDA和cuDNN就可以安装torch的GPU版本了。

torch安装

torch首页,下拉找到下图的位置,选择对应的平台,然后就会有对应的安装方式了,我使用的是anaconda,直接复制命令运行即可。


选择

https://www.jianshu.com/p/e38488cf1e91

Python量化投资网携手4326手游为资深游戏玩家推荐:《香肠派对下载

「点点赞赏,手留余香」

    还没有人赞赏,快来当第一个赞赏的人吧!
Matplotlib
0 条回复 A 作者 M 管理员
    所有的伟大,都源于一个勇敢的开始!
欢迎您,新朋友,感谢参与互动!欢迎您 {{author}},您在本站有{{commentsCount}}条评论