尽管已正确安装nvidia驱动程序,但无法导入tensorflow-gpu

时间:2019-04-02 08:36:11

标签: tensorflow python-3.6 rhel7 cudnn

我坚持让tensorflow-gpu在我的rhel7机器上工作。尽管已正确安装了nvidia驱动程序(我使用nvidia-smi命令对其进行了检查),但在尝试导入tensorflow时,却出现以下错误:

使用cuda 10.0

ImportError: libcudnn.so.7: cannot open shared object file: No such file or directory

使用cuda 10.1

ImportError: libcublas.so.10.0: cannot open shared object file: No such file or directory

libcublas.so.10.0存在,我在whereis中发现了它: /usr/lib64/libcublas.so.10

我尝试了2种不同的设置:

机器1

  • RHEL7
  • Python 3.6.3
  • CUDA 10.1
  • cuDNN 7
  • tensorflow-gpu 1.13.1

机器2

  • RHEL7
  • Python 3.6.3
  • CUDA 10.0
  • cuDNN 7
  • tensorflow-gpu 1.13.1

也尝试在两台机器上使用tensorflow-gpu 1.13.0。

按照各地报告的所有说明进行操作,以便正确安装所有内容,因此还请设置PATHLD_LIBRARY_PATH env变量。这里是一个例子:

PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}
LD_LIBRARY_PATH=/usr/lib64:${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

0 个答案:

没有答案