怎么判断gpu是否可用

在 Python 中，可以使用以下代码来检查 GPU 是否可用：

import torch

if torch.cuda.is_available():
    print('GPU is available')
else:
    print('GPU is not available')

如果输出结果为 `GPU is available`，则表示你的 GPU 可以被 PyTorch 使用。如果输出结果为 `GPU is not available`，则表示你的 GPU 无法被 PyTorch 使用，可以考虑使用 CPU 进行计算。