0
点赞
收藏
分享

微信扫一扫

pytorch1.4.0 CUDA11.0 python3.7安装记录


参考过程CUDA安装教程

CUDA教程2

找到自己电脑显卡的cuda版本

pytorch1.4.0 CUDA11.0 python3.7安装记录_python


CUDA是什么

版本是11.0.140

安装CUDA

11.1下载链接,但是我们不用这个

我们用的是11.0最新版的下载地址

下载选项设置(害,整整2个多G啊)。可以在下载按钮的地方右键,复制链接,然后在迅雷下面下载。虽然慢但是稳定。不过用Chrome复制下载地址的时候出了问题,改成edge,问题解决。

pytorch1.4.0 CUDA11.0 python3.7安装记录_CUDA_02


pytorch1.4.0 CUDA11.0 python3.7安装记录_python_03

双击运行

pytorch1.4.0 CUDA11.0 python3.7安装记录_CUDA_04


等待

pytorch1.4.0 CUDA11.0 python3.7安装记录_CUDA_05


同意并继续

pytorch1.4.0 CUDA11.0 python3.7安装记录_python_06


其实选精简安装就行了,但是还是不放心,还是选了自定义安装

pytorch1.4.0 CUDA11.0 python3.7安装记录_python_07


出现下面的界面

pytorch1.4.0 CUDA11.0 python3.7安装记录_下载地址_08


选什么装是个问题,第一次·和第二次装的时候,

全选的时候后面就出现Nsight Compute失败的情况

pytorch1.4.0 CUDA11.0 python3.7安装记录_python_09


于是取消了Nsight Compute,大不了以后手动装上这个模块也就是了

pytorch1.4.0 CUDA11.0 python3.7安装记录_python_10


然后,第三次重新装的时候又遇到杀毒软件杀掉的情况emmm

第四次装的时候遇到电脑重启,然后装的就GG

第五次决定全部按照默认配置来,失败默认路径是c盘,我把它换成d盘了

pytorch1.4.0 CUDA11.0 python3.7安装记录_下载地址_11


pytorch1.4.0 CUDA11.0 python3.7安装记录_python_12

然后就是等他安装完毕

pytorch1.4.0 CUDA11.0 python3.7安装记录_python_13


第六次换成了cuda11.0,还是因为nsight compute而安装失败

去掉Nsight· compute之后,安装成功

pytorch1.4.0 CUDA11.0 python3.7安装记录_CUDA_14


pytorch1.4.0 CUDA11.0 python3.7安装记录_python_15

检查是不是装好了nvcc,.exe

pytorch1.4.0 CUDA11.0 python3.7安装记录_CUDA_16


检查cupt的ddl是不是装上了

pytorch1.4.0 CUDA11.0 python3.7安装记录_CUDA_17


接下来安装cuDNN,CUDA的深度学习加速库

下载网址 选择对应的版本

pytorch1.4.0 CUDA11.0 python3.7安装记录_CUDA_18


pytorch1.4.0 CUDA11.0 python3.7安装记录_下载地址_19


检查一下cundnn的ddl有没有

pytorch1.4.0 CUDA11.0 python3.7安装记录_CUDA_20


将主目录重重命名为cudnn

pytorch1.4.0 CUDA11.0 python3.7安装记录_python_21


粘贴到cuda 的v11.0目录下

pytorch1.4.0 CUDA11.0 python3.7安装记录_下载地址_22

系统变量中的path项添加环境变量,已经有了两个cuda的路径,但是还缺两个

pytorch1.4.0 CUDA11.0 python3.7安装记录_CUDA_23

点浏览

pytorch1.4.0 CUDA11.0 python3.7安装记录_CUDA_24


添加这两个,点确定关闭

pytorch1.4.0 CUDA11.0 python3.7安装记录_下载地址_25


调出windows的cmd窗口,输入nvcc -V,检验cuda是否安装成功

pytorch1.4.0 CUDA11.0 python3.7安装记录_CUDA_26


下载pytorch1.4.0,windows,cu110没有对应版本的就很蛋疼

pytorch1.4.0 CUDA11.0 python3.7安装记录_CUDA_27


下载了这个版本

pytorch1.4.0 CUDA11.0 python3.7安装记录_下载地址_28


然后用anaconda的prompt来安装whl文件(我这边是已经新建了一个叫debayer的环境,没有的话得自己用anaconda建一个环境)

安装方法:pip install 路径+文件名+后缀

结果报错。查了一下,发现因为环境的python是3.7版本的

pytorch1.4.0 CUDA11.0 python3.7安装记录_CUDA_29


下载cp37版本的就ok看

pytorch的版本命名都是啥意思? 安装成功

pytorch1.4.0 CUDA11.0 python3.7安装记录_下载地址_30


测试一下,结果说没有装numpy模块

pytorch1.4.0 CUDA11.0 python3.7安装记录_python_31


用exit()命令退出python

pytorch1.4.0 CUDA11.0 python3.7安装记录_CUDA_32


添加清华源来pip numpy

import torchvision的时候报错

pytorch1.4.0 CUDA11.0 python3.7安装记录_CUDA_33


得安装torchvision

啥是torchvision?

我要用的是torchvision0.5.0

下载地址

pytorch1.4.0 CUDA11.0 python3.7安装记录_下载地址_34


torch和torchvision都装好了,可惜因为cuda和它们版本不匹配没办法用上GPU加速。等魔改完代码就可以用cuda跑了吧

pytorch1.4.0 CUDA11.0 python3.7安装记录_python_35


举报

相关推荐

0 条评论