美文网首页
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