【bug记录】RuntimeError: CUDA error: device-side assert triggered

RuntimeError: CUDA error: device-side assert triggered

我个人使用pytorch lightning时,报错RuntimeError: CUDA error: device-side assert triggered

我个人是因为分类类别数目对不上,但是在gpu上运行是不会提示错误的

解决方案:建议修改运行环境为cpu,就可以看到问题出在了哪里

trainer=pl.Trainer(devices=1,max_epochs=5,accelerator='gpu',default_root_dir='./checkpoints',limit_train_batches=5,limit_val_batches=1,limit_test_batches=5)
修改为
trainer=pl.Trainer(devices=1,max_epochs=5,accelerator='cpu',default_root_dir='./checkpoints',limit_train_batches=5,limit_val_batches=1,limit_test_batches=5)
可以清晰的看到自己的错误啦!

【bug记录】RuntimeError: CUDA error: device-side assert triggered_第1张图片

你可能感兴趣的:(bug)