位置: IT常识 - 正文

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

编辑:rootadmin
pytorch_lightning.utilities.exceptions.MisconfigurationException: You requested GPUs: 文章目录报错解决办法报错

推荐整理分享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]

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

笔者的报错代码:

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)
本文链接地址:https://www.jiuchutong.com/zhishi/290870.html 转载请保留说明!

上一篇:特卡波湖岸上的羽扇豆,新西兰 (© Stanislav Kachyna/Shutterstock)(特卡波小镇旅游攻略)

下一篇:奥拉基库克山国家公园中的塞夫顿山,新西兰南岛 (© AWL Images/Danita Delimont)(奥克拉库克湾)

免责声明:网站部分图片文字素材来源于网络,如有侵权,请及时告知,我们会第一时间删除,谢谢! 邮箱:opceo@qq.com

鄂ICP备2023003026号

网站地图: 企业信息 工商信息 财税知识 网络常识 编程技术

友情链接: 武汉网站建设 电脑维修 湖南楚通运网络