我想在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.176cudnn
:7.2tensorflow-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文件链接,但我不知道如何找到确切的问题并修复它。
我怎么解决这个问题?谢谢你的帮助!
cudnn找不到正确链接的原因是因为我使用的是anaconda
环境。我没有手动安装很多软件包,而是使用以下方法安装tensorflow-gpu
:
$conda create --name tf_gpu tensorflow-gpu
$source activate tf_gpu
这有效!本教程是here。
但我无法正确安装pip install tensorflow-gpu
的原因尚不清楚。还需要你的帮助和解释!