PyTorch 错误 RuntimeError: CUDA error: device-side assert triggered
训练数据的时候出现
RuntimeError:CUDA error:device-side assert triggeredCUDA kernel errors
might be asynchronously reported at some other API call, so the
stacktrace below might be incorrect.For debugging consider passing
CUDA_LAUNCH_BLOCKING=1.Compile withTORCH USE CUDA DSAto enable
device-side assertions.
检查一下数据集标签和训练时的标签是否一致。
原文地址：https://blog.csdn.net/qq_35793358/article/details/139088667
免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！