机器学习 ❀ RuntimeError:No CUDA GPUs are available

# python运行以检查安装是否正常
import torch
print(torch.cuda.is_available())

如果输出结果True,则正常,则可能是使用GPU时编号错误

打印出设备的数量,看看是不是选择的设备号错了

print(torch.cuda.device_count())

猜你喜欢

转载自blog.csdn.net/qq_42395917/article/details/126669711