yolov7 RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED
时间: 2023-11-05 21:56:07 浏览: 282
Error : Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so
当训练yolov7模型时出现RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED错误通常有两种解决思路。首先,你可以检查你的GPU服务器的CUDA版本,确保你安装了与你CUDA版本对应的Pytorch版本。其次,如果错误提示loss不能反传,你需要检查分类类别数量是否与实际数据的类别数量一致,特别是确保模型输出的关系类别数量定义正确。
阅读全文