RuntimeError: CuDNN错误: CUDNN_STATUS_SUCCESS

Question

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

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吗？如果是，是哪一个？

Answer 1

尝尝这个

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

Answer 2

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

torch.backends.cudnn.benchmark = True