我正在运行 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 吗？如果有，是哪一个？