首页 > 其他分享 >环境搭建-RuntimeError: CUDA error: device-size assert triggered

环境搭建-RuntimeError: CUDA error: device-size assert triggered

时间:2022-12-16 16:23:08浏览次数:46  
标签:triggered RuntimeError assert CUDA 显卡 size

报错如下图

 

 

后经测试,发现是因为启用了混合精度训练,导致了这种异常。去掉混合精度训练的参数,--fp16就可以了。

 

混合精度训练需要显卡支持,30X系列的显卡会支持,显卡版本低就不支持。

 

标签:triggered,RuntimeError,assert,CUDA,显卡,size
From: https://www.cnblogs.com/hanrp/p/16987667.html

相关文章