这是因为没有在pytorch中设置ddp (distributed data parallel) 就试图使用多块GPU。如果调试阶段不想用ddp,显式把使用的GPU数量设成1即可。
解决AttributeError: Can‘t pickle local object ‘ALIGNN.__init__.<locals>.<lambda>‘
最新推荐文章于 2024-05-27 10:21:59 发布
这是因为没有在pytorch中设置ddp (distributed data parallel) 就试图使用多块GPU。如果调试阶段不想用ddp,显式把使用的GPU数量设成1即可。