我原来的环境配置
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这一行,应用更改,完了重启就好了。

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

被折叠的 条评论
为什么被折叠?



