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

现在，我的问题：

该如何避免该错误？我是否需要将pip install用于cudnn的最新版本？如果是这样，哪个？

Answer 1

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

torch.backends.cudnn.benchmark = True

Answer 2

尝试

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

RuntimeError：CuDNN错误：CUDNN_STATUS_SUCCESS

2 个答案: