【问题标题】:When use tensorflow-gpu cudnn fails during the link time在链接时间内使用 tensorflow-gpu cudnn 失败时
【发布时间】:2019-03-03 07:36:32
【问题描述】:

我正在尝试在 ubuntu 上使用 tensorflow-gpu。经过一系列的安装,当我在python3中导入tensorflow时失败了。错误是

ImportError: /$MYPATH$/python3.6/site-packages/tensorflow/python/../libtensorflow_framework.so: 
symbol cudnnConvolutionBiasActivationForward version libcudnn.so.7 not defined in file libcudnn.so.7 with link time reference

我的环境是这样的:

  • cuda:9.0.176
  • cudnn:7.2
  • tensorflow-gpu:1.10

另外,我在 ~/.bashrc 中添加了 cuda 的路径:

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

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

我猜这个问题是由于错误的 cudnn 文件链接,但我不知道如何找到确切的问题并修复它。

我该如何解决这个问题?感谢您的帮助!

【问题讨论】:

    标签: python tensorflow anaconda


    【解决方案1】:

    cudnn 找不到正确链接的原因是因为我使用的是anaconda 环境。我没有手动安装很多包,而是简单地安装了tensorflow-gpu,使用:

    $conda create --name tf_gpu tensorflow-gpu
    $source activate tf_gpu
    

    这行得通!教程在here找到。

    但是为什么我可以不能使用pip install tensorflow-gpu正确安装的原因仍然未知。 仍然需要你的帮助和解释!

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 1970-01-01
      • 2017-07-17
      • 1970-01-01
      • 2019-02-20
      • 2023-02-06
      • 1970-01-01
      • 2017-02-10
      • 1970-01-01
      相关资源
      最近更新 更多