美文网首页
CUDA,cuDNN,pytorch安装(win)

CUDA,cuDNN,pytorch安装(win)

作者: 青椒rose炒饭 | 来源:发表于2019-10-29 20:50 被阅读0次

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

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,直接复制命令运行即可。

选择

相关文章

网友评论

      本文标题:CUDA,cuDNN,pytorch安装(win)

      本文链接:https://www.haomeiwen.com/subject/soogvctx.html