# python运行以检查安装是否正常
import torch
print(torch.cuda.is_available())
如果输出结果True,则正常,则可能是使用GPU时编号错误
打印出设备的数量,看看是不是选择的设备号错了
print(torch.cuda.device_count())版权声明:本文为qq_42395917原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。
# python运行以检查安装是否正常
import torch
print(torch.cuda.is_available())
如果输出结果True,则正常,则可能是使用GPU时编号错误
打印出设备的数量,看看是不是选择的设备号错了
print(torch.cuda.device_count())