深度学习 | Pytorch的GPU版本查看GPU是否可用、GPU版本、GPU数量

新建文件test.py复制如下内容运行即可。注意环境要选择正确

import torch

# 检查 GPU 是否可用
is_cuda_available = torch.cuda.is_available()
print(f"CUDA 可用: {
     is_cuda_available}")

if is_cuda_available:
    # 获取 GPU 数量
    gpu_count <