参考过程CUDA安装教程
CUDA教程2
找到自己电脑显卡的cuda版本
CUDA是什么
版本是11.0.140
安装CUDA
下载选项设置(害,整整2个多G啊)。可以在下载按钮的地方右键,复制链接,然后在迅雷下面下载。虽然慢但是稳定。不过用Chrome复制下载地址的时候出了问题,改成edge,问题解决。
双击运行
等待
同意并继续
其实选精简安装就行了,但是还是不放心,还是选了自定义安装
出现下面的界面
选什么装是个问题,第一次·和第二次装的时候,
全选的时候后面就出现Nsight Compute失败的情况
于是取消了Nsight Compute,大不了以后手动装上这个模块也就是了
然后,第三次重新装的时候又遇到杀毒软件杀掉的情况emmm
第四次装的时候遇到电脑重启,然后装的就GG
第五次决定全部按照默认配置来,失败默认路径是c盘,我把它换成d盘了
然后就是等他安装完毕
第六次换成了cuda11.0,还是因为nsight compute而安装失败
去掉Nsight· compute之后,安装成功
检查是不是装好了nvcc,.exe
检查cupt的ddl是不是装上了
接下来安装cuDNN,CUDA的深度学习加速库
下载网址 选择对应的版本
检查一下cundnn的ddl有没有
将主目录重重命名为cudnn
粘贴到cuda 的v11.0目录下
系统变量中的path项添加环境变量,已经有了两个cuda的路径,但是还缺两个
点浏览
添加这两个,点确定关闭
调出windows的cmd窗口,输入nvcc -V,检验cuda是否安装成功
下载pytorch1.4.0,windows,cu110没有对应版本的就很蛋疼
下载了这个版本
然后用anaconda的prompt来安装whl文件(我这边是已经新建了一个叫debayer的环境,没有的话得自己用anaconda建一个环境)
安装方法:pip install 路径+文件名+后缀
结果报错。查了一下,发现因为环境的python是3.7版本的
下载cp37版本的就ok看
pytorch的版本命名都是啥意思? 安装成功
测试一下,结果说没有装numpy模块
用exit()命令退出python
添加清华源来pip numpy
import torchvision的时候报错
得安装torchvision
啥是torchvision?
我要用的是torchvision0.5.0
torch和torchvision都装好了,可惜因为cuda和它们版本不匹配没办法用上GPU加速。等魔改完代码就可以用cuda跑了吧