loss.backward()报错RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED_loss.backward() runtimeerror: cuda error: cublas_s

作者：Gausst松鼠会 | 2024-02-16 01:04:15

踩

loss.backward() runtimeerror: cuda error: cublas_status_not_initialized when

最近在处理二分割问题时，遇到了标题所述的问题。

首先我可以确定我的环境绝对没问题。

仔细检查代码发现是网络用了深度监督机制，即输出多级结果和gt一同算loss。

仔细检查代码发现，几级输出有两级忘记用sigmoid激活直接算BCE Loss所以导致报错。

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/Gausst松鼠会/article/detail/89156