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/Do...
原创
2019-05-13 16:10:09 ·
18954 阅读 ·
20 评论