【发布时间】:2018-02-14 15:07:10
【问题描述】:
我已经安装了cuda-8.0 并使用以下方式安装了 TensorFlow:
sudo pip install tensorflow-gpu
当我尝试import tensorflow as tf 时,它说:
ImportError: libcublas.so.9.0: 无法打开共享对象文件:没有这样的文件或目录
我认为它是在 cuda-9.0 中搜索 libcublas
我在安装 cuda-8.0 之前清除了 cuda-9.0
有人可以提出其他可能的原因吗?
【问题讨论】:
-
您的 tensorflow 构建显然需要 CUDA 9.0。不知道为什么要删除它并安装 CUDA 8.0。您可能应该安装 CUDA 9.0,或切换到使用 CUDA 8.0 的旧版本 TF
-
如果您通过 pip 安装最新版本(1.5+),则需要 CUDA 9.0 或 9.1
-
我尝试安装 tensoflow 1.4,但仍然报同样的错误
-
Tensorflow 找不到库。因此,您应该确保它确实可用,并找出 TensorFlow 找不到它的原因。 Cuda 文档要求您设置一些环境变量,TensorFlow 也是如此(导出 LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/extras/CUPTI/lib64)。确保所有这些都设置正确。你也可以在
strace下运行python来查看它在哪里寻找库。
标签: tensorflow