【问题标题】:Is tensorflow-gpu (=<1.14) with CUDA & cuDDN supported in Red Hat Enterprise Linux 7/8 or cent os 7?Red Hat Enterprise Linux 7/8 或 cent os 7 是否支持带有 CUDA 和 cuDNN 的 tensorflow-gpu (=<1.14)?
【发布时间】:2020-01-26 21:23:37
【问题描述】:

我有带有 Nvidia 2080 超级显卡的 cent os 7,并且我安装了以下内容:

  1. 图形驱动视觉 430.4
  2. CUDA 10.1
  3. CuDNN 7.6
  4. conda env python 3.7
  5. tensorflow-gpu 1.14 (pip install tensorflow-gpu)

然后 pip 灌输代码所需的库。

但是当我尝试运行此代码 link 时,我得到 cuDNN 初始化错误..

非常感谢任何指南或建议 :-)

【问题讨论】:

标签: python-3.x tensorflow nvidia


【解决方案1】:

我不能说关于 CentOS,因为我从未安装过它,但在查看了您提供的信息之后。您图片中的错误说,Cudnn 无法初始化,这很可能是兼容性问题。您需要遵循these 的要求。 所以,彻底删除 Cuda 10.1 并安装 Cuda 10。看看问题是否仍然存在。

【讨论】:

    【解决方案2】:

    如果您使用的是 conda 环境,则应使用 conda 安装软件包。尝试使用 conda 安装 tensorflow-gpu;除了您已经拥有的 NVidia 驱动程序之外,它还应该处理您的所有依赖项。以这种方式做事允许 conda 处理其他包依赖项,例如 CUDA 工具包和 cudnn 等。

    来源:在 anaconda 环境中仅将 10.1 版本的 CUDA 工具包(带有驱动程序)安装到系统位置和 tensorflow-gpu,我在 CentOS 7 上安装了 tensorflow-gpu 1.14 和 Cuda 10.1 以及在 RHEL7 安装上。

    【讨论】:

      猜你喜欢
      • 2016-11-30
      • 2018-03-27
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2018-08-29
      • 2020-05-05
      • 2018-01-31
      相关资源
      最近更新 更多