以下命令来检查PyTorch和CUDA的版

在Python中，你可以使用以下命令来检查安装的PyTorch版本以及是否支持CUDA（如果已安装）：

# 首先导入torch库
import torch

# 检查PyTorch版本
print("PyTorch version:", torch.__version__)

# 如果有CUDA可用，会显示相关信息
if torch.cuda.is_available():
    print("CUDA is available.")
    # 获取当前可用的CUDA设备信息
    device = torch.cuda.current_device()
    print(f"CUDA device name: {torch.cuda.get_device_name(device)}")
    print(f"CUDA device capability: {torch.cuda.get_device_capability(device)}")
else:
    print("CUDA is not available.")

运行这段代码后，你会看到PyTorch的版本信息以及是否有CUDA支持，如果有，还会显示出使用的CUDA设备名称和能力。