目前看来
两个问题应该都是由于显存不足导致
解决方法一:
关闭多余占用显存的进程 (我同时开了两个就报第二个错)
解决方法二:
减小batch_size 从最小的往上加
或者在所有调用gpu的语句前后加上
torch.cuda.empty_cache()
释放多余缓存
例如:
改为
解决方法三:
换显卡 或者找网上的云计算平台训练(例如google colab)
目前看来
两个问题应该都是由于显存不足导致
解决方法一:
关闭多余占用显存的进程 (我同时开了两个就报第二个错)
解决方法二:
减小batch_size 从最小的往上加
或者在所有调用gpu的语句前后加上
torch.cuda.empty_cache()
释放多余缓存
例如:
改为
解决方法三:
换显卡 或者找网上的云计算平台训练(例如google colab)