很早之前使用的是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
至此解決問題。