报错如下图
后经测试,发现是因为启用了混合精度训练,导致了这种异常。去掉混合精度训练的参数,--fp16就可以了。
混合精度训练需要显卡支持,30X系列的显卡会支持,显卡版本低就不支持。
标签:triggered,RuntimeError,assert,CUDA,显卡,size From: https://www.cnblogs.com/hanrp/p/16987667.html
报错如下图
后经测试,发现是因为启用了混合精度训练,导致了这种异常。去掉混合精度训练的参数,--fp16就可以了。
混合精度训练需要显卡支持,30X系列的显卡会支持,显卡版本低就不支持。
标签:triggered,RuntimeError,assert,CUDA,显卡,size From: https://www.cnblogs.com/hanrp/p/16987667.html