# python运行以检查安装是否正常
import torch
print(torch.cuda.is_available())
如果输出结果True,则正常,则可能是使用GPU时编号错误
打印出设备的数量,看看是不是选择的设备号错了
print(torch.cuda.device_count())
# python运行以检查安装是否正常
import torch
print(torch.cuda.is_available())
如果输出结果True,则正常,则可能是使用GPU时编号错误
打印出设备的数量,看看是不是选择的设备号错了
print(torch.cuda.device_count())