windows cuda10.2 cudnn7.6.5下安装pytorch1.7.1 torchvision-0.8.2

pytorch版本的安装和cuda的版本是又对应关系的,如果版本不对应,是无法使用GPU进行工作的

win+r.输入cmd打开终端输入:nvcc -V或者(set cuda)可以查看自己的cuda版本,我这里安装的是cuda10.2,cudnn安装的为7.6.5

此环境下直接安装pytorch1.7.1,要么安装不成功,报没有pytorch1.7.1的版本,要么安装的是cpu版本的pytorch,可以通过
python
import torch
print(torch.version)的方式输出torch的版本是否是cpu版本的,如果是的话会输出torch 1.7.1+cpu,如果只输出的torch则人为安装的为GPU版本的,然后通过print(torch.cuda.is_available()),如果输出为True,表示安装成功,可以使用GPU工作,如果返回为false,则依然存在问题

pytorch安装
为避免上述问题,可以通过下载whl安装包的方式安装,链接https://download.pytorch.org/whl/torch_stable.html
请添加图片描述

根据cuda,python,系统版本去选择对应的pytorch版本和torchvision0.8.2版本,通过一下命令安装
pip install torch-1.7.1-cp36-cp36m-win_amd64.whl
pip install torchvision-0.8.2-cp36-cp36-win_amd64.whl

完成后,再次直行下面的命令进行查看
python
import torch
print(torch.version)
print(torch.cuda.is_available())
安装完成

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

风华正茂6666

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值