Pytorch GPU安装完毕后如何在python中导入？

您可以在 Python 中使用以下代码导入 PyTorch：

import torch

如果您的 PyTorch 安装在 GPU 上，您可以使用以下代码检查 GPU 是否可用：

if torch.cuda.is_available():
device = torch.device("cuda")
print("GPU is available")
else:
device = torch.device("cpu")
print("GPU is not available")