原因:
1 no CUDA-capable device ,可能原因时没有torch-GPU
2 在代码中指定了GPU(多卡情况下),而单卡中是 ‘0’
os.environ["CUDA_VISIBLE_DEVICES"] = '1'
解决:
1 将 os.environ[“CUDA_VISIBLE_DEVICES”] = ‘1’ 修改为
os.environ["CUDA_VISIBLE_DEVICES"] = '0'
3 或者注释该行代码
最后,附上pytorch自查代码:
import torch
import torchvision
print(torch.__version__)
print(torch.cuda.is_available())
print(torch.cuda.device_count())
print(torch.cuda.get_device_name())
print(torch.cuda.current_device())