本来不怎么想搞这个的,但是想一想百度到的都是一年前的内容,而且安装还比较复杂,所以决定还是写一下吧。当然现阶段我还不知道我配置的这个有什么问题。
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安装
下载地址
这个需要登录,填写问卷调查之后才能下载。
只需要将里面的文件对应拷贝到CUDA安装目录下对应的位置即可。
这样子CUDA就全部安装完成了,接下来就可以安装pytorch的GPU版本了。如果上面的步骤没有做就会遇到torch导入失败的情况,虽然torch安装的时候也是一切顺利。
torch导入失败
装完上面的CUDA和cuDNN就可以安装torch的GPU版本了。
torch安装
torch首页,下拉找到下图的位置,选择对应的平台,然后就会有对应的安装方式了,我使用的是anaconda,直接复制命令运行即可。
网友评论