这个屌丝很懒，什么也没留下！

pytorch lightning报错all tensors to be on the same device

作者：正经夜光杯 | 2024-07-26 20:33:44

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!

trainer = pl.Trainer(max_epochs=train_params.iterations, logger=tb_logger,
                         accelerator='gpu', devices=1)
1
2

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/正经夜光杯/article/detail/887088