TensorFlow pycharm 中的 GPU 警告

Posted 2023-02-23

【中文标题】TensorFlow pycharm 中的 GPU 警告

【英文标题】：Warning for GPU in tensorflow pycharm

【发布时间】：2021-01-31 22:38:18

【问题描述】：

我最近在安装 Tensorflow 时遇到了问题，但我通过在 anaconda navigator 中创建虚拟环境解决了这个问题。我收到警告说

2020-10-17 13:44:04.120482：W tensorflow/stream_executor/platform/default/dso_loader.cc:55] 无法加载动态库“cudart64_101.dll”； dlerror: 未找到 cudart64_101.dll 2020-10-17 13:44:04.120745: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] 如果您的机器上没有设置 GPU，请忽略上面的 cudart dlerror。

问题是我的笔记本电脑 (T2000) 上有一个 GPU。我该如何解决这个问题？

【问题讨论】：

您使用的是什么版本的 TF 和 CUDA？

我使用的是 tensorflow 2.1.o 而我的 CUDA 是 10.2

【参考方案1】：

您是否使用 pip 安装了 tensorflow？如果您确实卸载它并使用 conda 安装 tensorflow。 Conda 自动安装 cudnn 7.6.5 和 CUDA Toolkit 10.1.243 和 tensorflow 2.1.0。如果您想要 tensorflow 2.2，请按照说明首先安装 2.1.0。然后使用 pip 安装 2.2 和 pip install tensorflow ==2.2.0。 2.2 与随 2.1 一起安装的工具包和 cudnn 版本兼容。 Conda 只能安装 tensorflow 到 2.1.0 版本。使用 pip 或 conda 安装您需要的任何其他软件包。

【讨论】：

我是通过创建自己的环境来安装的。然后我在环境中使用 conda 安装了 tensorflow。然后我用它作为我在 pycharm 中的解释器。我也使用了 pip，但它不允许我使用 tensorflow 给我这个错误，“ImportError: DLL load failed: The specified procedure could not be found.”

你是先用conda安装tensorflow，然后用pip安装tensorflow 2.2？