import torch
print(torch.cuda.is_available())
返回结果是True,则PyTorch的GPU安装成功
本文介绍了一种简单的方法来检查PyTorch是否已成功安装并配置了GPU支持。通过运行Python代码,如果返回结果为True,则表明PyTorch能够访问GPU资源。
import torch
print(torch.cuda.is_available())
返回结果是True,则PyTorch的GPU安装成功
1229
3万+
3万+

被折叠的 条评论
为什么被折叠?