这几天在跑代码时竟然报错说没有cuda,这这这,不敢相信机器的报错信息。我在cmd命令行验证了一下:
import torch
print(torch.cuda.is_available())
结果显示果然是False,之前有windows10+caffe环境成功配置的经验,对环境的配置还是比较自信的。因此撸起袖子就干啊,根据报错的信息,我猜测是电脑系统的升级对显卡的驱动有一定的影响或者是后续网络的环境搭建破坏掉了原有的环境。
首先在cmd的验证信息:
我对NVIDIA重新安装了一下,cuDNN也替换了CUDA的三项内容,但最终显示出的结果与上图一样,我使用了360驱动大师进行驱动的安装,仍旧不见效。因为我在安装CUDA的时候并没有使用安装nvidia的驱动,所以我在去官网下载了驱动,链接为:NVIDIA驱动官网下载
选择自己电脑相应的产品(我电脑是GTX 1650),强烈推荐使用迅雷下载,将该网址复制到迅雷,下载速度快的飞起,不然网页下载需要等得让你打瞌睡。
下载完成后,安装该驱动,安装步骤和cuda安装一样,选择精简安装即可,不需要添加系统环境变量。最后,再次验证:
完美解决!
版权声明:本文为qq_33538651原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。