pytorch_lightning.utilities.exceptions.MisconfigurationException: You requested GPUs: [1] But…

pytorch_lightning.utilities.exceptions.MisconfigurationException: You requested GPUs: [1] But your machine only has: [0]

笔者的报错代码:

trainer = Trainer(
                        max_epochs= config.max_epochs,
                        gpus= [fix_config.hparams.gpus],
                        distributed_backend= fix_config.hparams.distributed_backend,
                        benchmark= fix_config.hparams.benchmark,
                        deterministic= fix_config.hparams.deterministic,
                        logger=logger,
                        callbacks=[
                            early_stop_callback,
                            checkpoint_callback
                        ]
                      )

if __name__ == "__main__":

    parser = ArgumentParser()
    parser.add_argument("--gpus", default=1, type=int)
    ...

这是使用 pytorch_lightning框架时遇到的错误。报错说,你想要的 GPUs数量为 [1],但是你的机器只有 [0]个。

看到这个,相信大多数人的第一反应是:

  • 欸!报错说我没有gpu!
  • 接着去检查一下cuda
    pytorch_lightning.utilities.exceptions.MisconfigurationException: You requested GPUs: [1] But...
    注:如果显示cuda为Flase,则表示你 没有装cuda或者 torch装的是cpu版本,需要重新装一下。笔者在这篇笔记里介绍了配置cuda与相应gpu版本的pytorch、torchvision、torchaudio的辛酸历程,可以作为参考。
  • cuda是可以用的!所以它为什么说我只有 [0]呢?

这样一来,重新翻译一下错误: 你想要2个GPU,但是你的机器只有1个GPU。

把上述代码中, gpu数量改为 0

...

parser.add_argument("--gpus", default=0, type=int)

Original: https://blog.csdn.net/qq_44250700/article/details/125453932
Author: Begonia_cat
Title: pytorch_lightning.utilities.exceptions.MisconfigurationException: You requested GPUs: [1] But…

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/718481/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球