我已经根据 https://www.tensorflow.org/install/install_linux 上的说明安装了 Cuda Toolkit 9.0 和 cuDNN v7.0 (和 nvidia 网站)说明。但是,当我尝试使用 tensorflow 运行一些脚本时,它说找不到它
ImportError: libcublas.so.9.0: 无法打开共享对象文件: 没有那个文件或目录
即使我的 LD_LIBRARY_PATH 设置正确:
回显 $LD_LIBRARY_PATH
/usr/local/cuda-9.0/lib64:/usr/local/cuda-9.0/extras/CUPTI/lib64
有什么我想念的吗?
最佳答案
如果您使用运行文件安装方法安装了 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
)。
关于即使设置了 LD_LIBRARY_PATH,Tensorflow 也找不到 libcublas.so.9.0,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/49401835/