解决win10下安装cuda9.0在pytorch中torch.cuda.is_available()返回False的问题

很早之前使用的是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

至此解决问题。

你可能感兴趣的:(Python,CUDA,Pytorch)