当前位置:   article > 正文

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

error occurred when executing ade_animatediffloaderwithcontext:

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了

  1. def build_model(self):
  2. self.net_bone = build_model(base_model_cfg) # model.build_model, base_model_cfg = 'resnet'
  3. print('=>1 self.config.cuda:', self.config.cuda) # todo =>cuda?//
  4. # self.config.cuda = False
  5. print('=>11 self.config.cuda:', self.config.cuda) # todo =>cuda?//
  6. '''if self.config.cuda:
  7. self.net_bone = self.net_bone.cuda()'''
  8. self.net_bone.eval() # use_global_stats = True # todo =>?//
  9. self.net_bone.apply(weights_init)

 

 

 

 

 

 

 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/不正经/article/detail/370398?site
推荐阅读
相关标签
  

闽ICP备14008679号