Pytorch/Torch：电脑无cuda情况下，使用CPU运行

显示错误“Torch not compiled with CUDA enabled”

修改方法

第一步

先在程序最开始加上

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

    
    

    
    

然后把所有的

.cuda()

    
    

    
    

改成

.to(device)

    
    

    
    

如果并未解决，请进行第二步，若解决，请忽视

第二步

把所有

torch.load(model_path)

    
    

    
    

改为

torch.load(model_path,map_location='cpu')

    
    

    
    

一个小例子

train_data = torch.from_numpy(train_data_np).cuda()
···
model.load_state_dict(torch.load(model_path)

    
    

    
    

修改为

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
···
train_data = torch.from_numpy(train_data_np).to(device)
model.load_state_dict(torch.load(model_path,map_location='cpu'))

    
    

    
    

转载自

这是把两篇博客的内容拼接在一起，由于转载不能填两个链接，所以厚着脸皮写成原创了
https://blog.csdn.net/quantum7/article/details/89333293

https://blog.csdn.net/ssjdoudou/article/details/103640129?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-1.nonecase&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-1.nonecase