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

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)

 

 

 

 

 

 

 

你可能感兴趣的:(显著性检测,EGNet)