pytorch - RuntimeError：CuDNN 错误：CUDNN_STATUS_SUCCESS

Question

我正在运行从 github 下载的 UI 代码。它应该可以工作（我看到其他人设法激活它）。当我尝试运行它时，我收到以下错误消息：

RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESS

该代码使用 pytorch 0.4.1。我已经安装了 cuda。

当我运行命令时，cat /usr/local/cuda/version.txt 我得到了答案：

CUDA Version 10.0.130

当我运行命令时，conda list -n <my env name> 我看到：

cudatoolkit ver 9.0
cudnn       ver 7.6.5

现在，我的问题：

我应该怎么做才能避免这个错误？我需要使用pip install更新版本的 cudnn 吗？如果有，是哪一个？

标签： pytorchcudnn

Answer 1

我也面临同样的问题。在我的例子中，PyTorch 版本是 0.4.1，Cuda 版本是 9.0。我通过添加这段代码解决了这个问题：

torch.backends.cudnn.benchmark = True