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

Question

我已经根据 https://www.tensorflow.org/install/install_linux(和nvidia站点(指令安装了Cuda Toolkit 9.0和cuDNN v7.0。但是，当我尝试使用张量流运行一些脚本时，它说找不到它

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

我错过了什么吗？

Answer 1

如果您已经使用 runfile 安装方法安装了 cuda，您还必须根据 Nvidia 文档设置 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(。