Tensorflow-gpu(= <1.14)在Red Hat Enterprise Linux 7/8或cent os 7中是否支持CUDA和cuDDN?

问题描述 投票:-1回答:2

我有配备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安装tensorflow-gpu)

然后点入为代码填充所需的库。

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

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

enter image description here

python-3.x tensorflow nvidia cudnn
2个回答
0
投票
我不能说CentOS,因为我从来没有安装过CentOS,而是在查看了您提供的信息之后。您图片中的错误说,Cudnn无法初始化,这很可能是兼容性问题。您需要遵循these要求。因此,完全删除Cuda 10.1并安装Cuda10。查看问题是否仍然存在。

0
投票
如果使用的是conda环境,则应使用conda安装软件包。尝试使用conda安装tensorflow-gpu;除了已经拥有的NVidia驱动程序外,它还应照顾所有的依赖项。通过这种方式,conda可以处理其他程序包依赖项,例如CUDA工具包和cudnn。
© www.soinside.com 2019 - 2024. All rights reserved.