pytorch1.4.0怎么区分是cpu还是gpu
时间: 2024-05-23 22:15:27 浏览: 132
torch-1.4.0+cpu-cp38-cp38-win_amd64.whl
如果您在安装 PyTorch 1.4.0 时选择了 CUDA 支持,那么您就可以利用 GPU 进行加速。如果您没有选择 CUDA,则 PyTorch 将默认在 CPU 上运行。您可以通过调用 `torch.cuda.is_available()` 函数来检查是否可用 GPU。如果返回 True,则代表您的系统支持 GPU,否则只能在 CPU 上运行。
阅读全文