如何查看torch是cpu版本还是gpu版本

你可以通过以下代码来查看你的torch是CPU版本还是GPU版本：

import torch

if torch.cuda.is_available():
    print("Torch is using GPU.")
else:
    print("Torch is using CPU.")

如果你的输出结果中显示 "Torch is using GPU."，则表示你的torch是GPU版本；如果输出结果中显示 "Torch is using CPU."，则表示你的torch是CPU版本。