tensorflow - 即使设置了LD_LIBRARY_PATH，Tensorflow也找不到libcublas.so.9.0

即使设置了LD_LIBRARY_PATH，Tensorflow也找不到libcublas.so.9.0

时间：2018-03-21 08:54:32

标签： tensorflow

我根据https://www.tensorflow.org/install/install_linux（和nvidia site）指令的说明安装了Cuda Toolkit 9.0和cuDNN v7.0。但是，当我尝试使用tensorflow运行某些脚本时，它说它无法找到它

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

即使我的LD_LIBRARY_PATH设置正确： echo $LD_LIBRARY_PATH /usr/local/cuda-9.0/lib64:/usr/local/cuda-9.0/extras/CUPTI/lib64

有什么我想念的吗？

1 个答案:

答案 0 :(得分：1)

如果您使用runfile安装方法安装了cuda，则必须根据Nvidia documentation设置PATH变量。我建议您将export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}和export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64 ${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}（64位）或export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib ${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}（32位）添加到.bashrc（如果您使用bash）。