【问题标题】: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,并且我安装了以下内容:
- 图形驱动视觉 430.4
- CUDA 10.1
- CuDNN 7.6
- conda env python 3.7
- 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 安装上。