win10+cuda9.1+cudnn7.1.3+pytorch-0.4.1

win10+cuda9.1+cudnn-9.1-windows10-x64-v7.1.3+pytorch-0.4.1-py37_cuda90_cudnn7he774522_1

新手第一次安装,竟然安装了一天,醉了醉了。

写在最前面,安装第三步安装pytorch和torchvision时,最后是用conda新建一个虚拟环境,防止破坏原始的python环境条件,用conda创建一个虚拟环境对于以后做项目来讲,不需要有无用的包造成影响等。

1、上一篇文章说了,先看自己电脑对应的cuda版本型号,这里就不多说了。
直接上连接点击就直接跳转cuda。找到自己需要的版本直接下载就行。
安装的时候默认傻瓜式操作就可以,中间值勾选第一个就可以,其余的不需要。
2、第二步就是下载cudnn了,我的电脑是对应的7.1.3,当然只要他对应的是cuda是9.1版本就行,但是官网更新之后就没有9.1版本的了,可以选择对应版本是9.0的,向下兼容。跳转连接cudnn这个需要先注册,注册之后才能下载。
3.完成上面两部分基本上就完成三分之一了。安装pytorch的难处在于不知道版本匹配,得自己去弄明白,我也是瞎碰上了。
pytorch我安装的是pytorch-0.4.1gpu版本的,我一开始安装的是pytorch-1.1.0版本的,但是这种不知道为什么老是显示找不到模块,看了好多博客怎么改,都没有改好,结果换了一个版本,就行了。
pytorch的安装,我是用的conda安装,这样的话就不需要下载whl文件,我是用的tar.bz2文件,在官网下载太麻烦了,直接在用清华镜像下载。
镜像地址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
对应的Torchvision也要一块下载安装,这个版本也不要太高,我选的是torchvision-0.3.0-py37-cuda90
然后就是安装了

没有对应的cuda9.1版本的时候,就选择对应的9.0的,这样一般不会出错。

// 一定要到文件相应的目录,在执行命令。
conda install --offline pytorch-0.4.1-py37_cuda90_cudnn7he774522_1.tar.bz2
conda install --offline torchvision-0.3.0-py37_cu90_1.tar.bz2

这样之后应该就是安装好了。

验证是否成功

import torch  //不报错就代表成功了

//验证是否可以使用GPU和CUDA
import torch
torch.cuda.is_available()//返回True就行了

当然这里也可以用pip install xxxx(对应的pytorch的whl文件)
这里只附上下载地址,安装方法基本是一样的。
https://download.pytorch.org/whl/cu90/torch_stable.html

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值