很早之前使用的是Tensorflow框架,安装了cuda 9.0/cudnn/tensorflow也一直能用GPU训练。现在直接下载pytorch-cuda90版本的安装后使用torch.cuda.is_available()却返回的是false。了解了一下是显卡驱动太旧了,更新一下即可。
虽然在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin文件夹中使用命令nvcc -V能够查看到cuda已成功安装,但是并不能调用GPU
打开Geforce Experience,在“驱动程序”中更新最新的显卡驱动程序,由原来的398.xx更新到436.XX
>>> import torch
>>> torch.cuda.is_available()
True
至此解决问题。