首页IT科技pytorchlightning 进度条控制(pytorch_lightning.utilities.exceptions.MisconfigurationException: You requested GPUs: [1] But…)

pytorchlightning 进度条控制(pytorch_lightning.utilities.exceptions.MisconfigurationException: You requested GPUs: [1] But…)

时间2025-05-01 22:19:17分类IT科技浏览6764
导读:报错 pytorch_lightning.utilities.exceptions.MisconfigurationException: You requested GPUs: [1] But your machine only has: [0]...

报错

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]呢?

笔者最开始也想不通       ,查阅了少得可怜的资料后                    ,发现这个作者的猜想可能是对的:也就是说              ,gpu数量是从0开始计数的!

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

解决办法

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

... parser.add_argument("--gpus", default=0, type=int)
声明:本站所有文章,如无特殊说明或标注             ,均为本站原创发布       。任何个人或组织                     ,在未征得本站同意时       ,禁止复制      、盗用              、采集                     、发布本站内容到任何网站      、书籍等各类媒体平台                     。如若本站内容侵犯了原著者的合法权益             ,可联系我们进行处理             。

创心域SEO版权声明:以上内容作者已申请原创保护,未经允许不得转载,侵权必究!授权事宜、对本内容有异议或投诉,敬请联系网站管理员,我们将尽快回复您,谢谢合作!

展开全文READ MORE
react路由守护(React路由) seo关键词排名工具有哪些(解密SEO关键词排名注册价格:让您的网站跻身榜首)