print(torch.cuda.is_available())为False,训练无法使用GPU的情况怎么办

这种情况是因为cuda与torch版本不一致情况导致的,

CUDA 10.0

https://download.pytorch.org/whl/cu100/torch_stable.html

CUDA 9.0

https://download.pytorch.org/whl/cu90/torch_stable.html

我安装了cuda10的torch1.2后问题解决!

猜你喜欢

转载自blog.csdn.net/com_fang_bean/article/details/107295107