在链接时使用tensorflow-gpu cudnn失败时

问题描述 投票:0回答:1

我想在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文件链接,但我不知道如何找到确切的问题并修复它。

我怎么解决这个问题?谢谢你的帮助!

tensorflow linker anaconda cudnn
1个回答
0
投票

cudnn找不到正确链接的原因是因为我使用的是anaconda环境。我没有手动安装很多软件包,而是使用以下方法安装tensorflow-gpu

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

这有效!本教程是here

但我无法正确安装pip install tensorflow-gpu的原因尚不清楚。还需要你的帮助和解释!

© www.soinside.com 2019 - 2024. All rights reserved.