检查pytorch配置

使用下面的代码进行测试

CUDA

import torch

# 打印PyTorch版本
print(torch.__version__)

# 检查是否可以使用CUDA（GPU）
print(torch.cuda.is_available())

MPS

import torch

# 打印PyTorch版本
print(torch.__version__)

# 检查MPS是否可用
if torch.backends.mps.is_available():
    mps_device = torch.device("mps")
    print("MPS device is available.")
else:
    print("MPS device is not available.")