RuntimeError：CuDNN 错误：CUDNN_STATUS_SUCCESS答案

【问题标题】：RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESSRuntimeError：CuDNN 错误：CUDNN_STATUS_SUCCESS
【发布时间】：2020-07-06 08:56:26
【问题描述】：

我正在运行从 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

现在，我的问题：

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

【问题讨论】：

您使用的是哪种 GPU 设备？

标签： pytorch cudnn

【解决方案1】：

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

torch.backends.cudnn.benchmark = True

【讨论】：

【解决方案2】：

试试这个

if torch.cuda.is_available():
    device = torch.device("cuda")
    print("working on gpu")
else:
    device = torch.device("cpu")
    print("working on cpu")

【讨论】：