PyTorch GPU 安装后是否默认使用 GPU 训练模型?
是的,安装了 GPU 版本的 PyTorch 后,如果你在代码中使用了 GPU 相关的语句,PyTorch 会默认使用 GPU 来跑模型。但是如果你的代码中没有使用 GPU 相关的语句,那么 PyTorch 会默认使用 CPU 来跑模型。你可以使用以下语句来检查当前使用的设备:
import torch
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
print(device) # 输出当前设备
如果输出结果是 'cuda',表示当前在使用 GPU;如果输出结果是 'cpu',表示当前在使用 CPU。
原文地址: https://www.cveoy.top/t/topic/lyvK 著作权归作者所有。请勿转载和采集!