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),问题解决
最新显卡驱动程序安装途径:
- https://www.nvidia.cn/Download/index.aspx?lang=cn 手动选择本机配置,下载安装即可
- https://www.nvidia.cn/geforce/geforce-experience/ 利用Nvidia GeForce Experience 自动更新最新显卡驱动