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
注:如果显示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/
转载文章受原作者版权保护。转载请注明原作者出处!