解决Pytorch的版本为CPU，导致无法使用CUDA进行GPU加速计算的问题

最新推荐文章于 2025-06-01 15:42:12 发布

DivinerNJ

最新推荐文章于 2025-06-01 15:42:12 发布

阅读量936

点赞数 3

CC 4.0 BY-SA版权

文章标签： pytorch 人工智能 python

本文链接：https://blog.csdn.net/m0_53420264/article/details/137894219

首先通过语句，得到自己的pytorch版本：

import torch
print(torch.__version__)

如果+号后面是cu和一串数字，那么恭喜，你的pt版本支持CUDA，那么你可以通过以下代码获得你的CUDA设备编号：

import torch

# 检查系统上可用的 CUDA 设备数量
device_count = torch.cuda.device_count()
print("可用的 CUDA 设备数量:", device_count)

# 遍历每个 CUDA 设备，并打印设备编号和设备名称
for i in range(device_count):
    device_name = torch.cuda.get_device_name(i)
    print("CUDA 设备 {}: {}".format(i, device_name))

如果+号后面是“cpu”，那么请遵循以下步骤：