当前位置: 代码迷 >> 综合 >> RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESS
  详细解决方案

RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESS

热度:63   发布时间:2023-11-21 11:28:07.0

我的PyTorch version 0.4.1,
cudatoolkit 9.0
cudnn 7.6.5
在程序中插入下面这句话,很好地解决了!至于更深入的原因目前不是很清楚,其中解决方案来源于
RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESS

torch.backends.cudnn.benchmark = True

  相关解决方案