遇到问题EGNet(3)：cpu版本运行gpu版本的pytorch, 报错:AssertionError:torch not compiled with cuda enabled(成功运行)

计算机视觉-Archer

于 2019-09-25 23:18:39 发布

阅读量3.6w

点赞数 2

分类专栏：显著性检测-合并进去 EGNet

本文链接：https://blog.csdn.net/zjc910997316/article/details/101394648

版权

EGNet 同时被 2 个专栏收录

45 篇文章

订阅专栏

显著性检测-合并进去

22 篇文章

订阅专栏

本文详细介绍了如何使用PyTorch加载模型，并解决了在不同设备上加载模型时遇到的问题，尤其是当从GPU设备加载到CPU设备时的解决策略。文章提供了正确的torch.load()函数调用方式，确保模型能在没有GPU的环境下正常运行。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

https://www.cnblogs.com/xiaodai0/p/10413711.html
这个博客建议个格式
torch.load("0.9472_0048.weights",map_location='cpu')

下面格式成功运行

torch.load(xxx, map_location=torch.device('cpu'))  # 这是终端报错提示我使用的格式

不管读的什么模型都加上，就没问题了

torch.load(self.config.pre_trained, map_location=torch.device('cpu'))

1 solver.py 这里很奇怪,默认是True虽然我 --cuda False了

2 parser.add_argument('--cuda', type=bool, default=False) 这里也默认是False了

    def build_model(self):

        self.net_bone = build_model(base_model_cfg)  # model.build_model, base_model_cfg = 'resnet'

        print('=>1  self.config.cuda:', self.config.cuda)  # todo =>cuda?//
        # self.config.cuda = False
        print('=>11 self.config.cuda:', self.config.cuda)  # todo =>cuda?//
        '''if self.config.cuda:  
            self.net_bone = self.net_bone.cuda()'''
        self.net_bone.eval()  # use_global_stats = True  # todo =>?//
        self.net_bone.apply(weights_init)