升级pytorch-gpu

本文详细记录了作者在Windows环境下,从PyTorch 1.2.0 + CUDA 10.0 升级到 PyTorch 1.7.0 的过程,包括自查CUDA版本、更新显卡驱动,以及通过不同的安装方法下载和安装PyTorch GPU版本。

我原来的环境配置

Windows + Anaconda3 + PyTorch=1.2.0 + torchvision=0.4.0 + cuda=10.0 + cudnn=7.4.1.5

升级缘由

找了个代码,pytorch要求1.7,so开始折腾起来了

曲折的过程

本来挺简单,因为眼瞎,走了很多弯路

正题

1、终端自查

(win+r,输出cmd)

nvcc -V,显示正常,说明cuda在,版本是10.0.130

nvidia-smi,显示正常,说明显卡驱动在,

 网上查找,显卡驱动版本与cuda版本是否匹配

cuda10.0.130必须显卡驱动大于411.31,我是的462.42,没有问题。如果不匹配,就要升级显卡驱动。

(找到了一个Linux的设置)设置->软件更新->附加驱动->选择nvidia最新驱动(361)->应用更改,选择435这一行,应用更改,完了重启就好了。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值