Win10下cuda10.1安装正确torch.cuda.is_available()为False的解决方法

在安装完cuda10.1之后,cmd下执行nvcc -V,可以正确显示cuda版本号,
但使用pytorch时,执行torch.cuda.is_available()却得到False

原因:
我机子的显卡驱动版本太低,为398.xx,不支持cuda10.1

解决:
更新显卡驱动至当前最新版本(430.xx),问题解决

Win10下cuda10.1安装正确torch.cuda.is_available()为False的解决方法_第1张图片
最新显卡驱动程序安装途径:

  1. https://www.nvidia.cn/Download/index.aspx?lang=cn 手动选择本机配置,下载安装即可
  2. https://www.nvidia.cn/geforce/geforce-experience/ 利用Nvidia GeForce Experience 自动更新最新显卡驱动

你可能感兴趣的:(pytorch,cuda)