无法在 Anaconda 环境中使用 GPU答案

【问题标题】：Unable to use GPU in Anaconda environment无法在 Anaconda 环境中使用 GPU
【发布时间】：2020-11-13 07:58:58
【问题描述】：

我想在 Linux 上使用 GPU 和 Anaconda 环境。我应该已经调整了每个模块的版本，但它不起作用。使用 conda 安装 Cuda 和 cuDNN。下面列出了各个模块和驱动的版本：

・GPU：RTX 2070 SUPEER ・操作系统：Linux Mint 19.3 Tricia ( Ubuntu 18.04 ) ・Nvidia驱动：435.21

# conda list tensorflow
tensorflow                2.1.0           gpu_py37h7a4bb67_0
tensorflow-base           2.1.0           gpu_py37h6c5654b_0
tensorflow-estimator      2.1.0              pyhd54b08b_0
tensorflow-gpu            2.1.0                h0d30ee6_0

# conda list cudnn
cudnn                     7.6.5                cuda10.1_0

# conda list cudatoolkit
cudatoolkit               10.1.243             h6bb024c_0

我可以通过输入以下命令看到GPU

from tensorflow.python.client import device_lib
device_lib.list_local_devices()

当我运行训练脚本时，出现以下错误

tensorflow.python.framework.errors_impl.UnknownError:  Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
         [[node conv1d_3/convolution ......

如何让它正常工作？

【问题讨论】：

你是如何安装cudnn的？这可能与版本与系统上现有软件包之间的不兼容有关。如果可能，请描述您是如何安装tensorflow GPU 版本的。
感谢您的回复！我使用以下命令安装了 tensorflow/cuda/cudnn conda install tensorflow=2.1.0=gpu_py37h7a4bb67_0。
您想特别使用版本tensorflow=2.1.0 还是只想让它运行？您能否尝试创建一个新环境（例如conda create --name newENV python=3.7，我希望您使用python3）并尝试conda install tensorflow-gpu，让我知道tensorflow 是否能够识别GPU。
我只是想让它运行。当我按照您教我的方式进行尝试时，GPU 会显示以下命令（'device_lib.list_local_devices()'），但在运行脚本时出现相同的错误。安装 keras-gpu 或 scikit-learn 有负面影响吗？
为了调试这个问题，我强烈建议你： 1. 创建一个新的 conda 环境 2. 执行conda install tensorflow-gpu 的全新安装 3. 运行这个来测试：tensorflow.config.list_physical_devices("GPU") 如果错误仍然存在发生了，我认为你的包在conda之外有问题。

标签： python tensorflow gpu

【解决方案1】：

根本原因：缺乏硬件资源。

解决方法： 新装了TF 2.0，跑了一个简单的Minst教程，还好，打开另一个notebook，尝试运行，遇到了这个问题。

我退出了所有笔记本并重新启动了 Jupyter，只打开了一个笔记本，运行成功。问题似乎是内存或在 GPU 上运行多个笔记本

更多阅读here。

【讨论】：