RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED
RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED
训练模型时出现上面的错误,同样的代码之前运行都没有问题,突然出现这个错误。查看gpu信息时发现显卡都被占用了,显存接近上限,后来在显存还剩一半时重新运行代码就没有出现这个错误了。因此,这种问题可能是显存不够导致的。害的我还在重装cudnn,巨坑!!!
...
原创
2020-12-07 15:30:27 ·
2022 阅读 ·
2 评论