在使用pytorch_lightning进行模型训练的时候,报了如下的错误
请教大佬之后在此记录一下下
解决方法
设置为gpus=0
原理
原本调用train的参数设置的是gpus=1,字面理解就是使用一个gpu去跑,对应到配置上就是,需要一个cuda
但是此时我的环境没配cuda,所以相当于我没有gpu,所以必须更改gpus=0,即,换成cpu去跑程序。
也有一种解释是,我的本机上只有一个gpu,而gpu的编号是从0开始,所以gpus=1表示使用的是第二个gpu,所以需要把编号改成0,才表示使用自己的gpu跑。
可能理解不太对,所以记录在这里,希望评论大哥能解答一下哪种理解正确哇(或者都不正确?)