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

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

import torch

如果您已经安装了 GPU 版本的 PyTorch，它将自动使用 GPU 运行。如果您想确保它正在使用 GPU，请使用以下代码：

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print(device)

这将输出您的设备名称，如果您的设备支持 CUDA，则应该输出 "cuda"。