使用conda安装tensorflow-gpu并给我一个错误

时间:2019-03-01 19:55:11

标签: tensorflow anaconda

我安装了最新的NVIDIA驱动程序。我安装了cuda工具包9.0,并在.bashrc中设置了路径。我已经使用以下命令conda install -c anaconda tensorflow-gpu = 1.5.0安装了conda tensorflow-gpu。下载了每个需要的软件包,当我尝试编译一个简单的tensorflow测试程序时,出现以下错误“ ImportError:libcuda.so.1:无法打开共享库文件:没有这样的文件或目录”。我在安装tensorflow-gpu时检查了cudatoolkit是9.0,而cudNN是7.3.1。我之前已经进行了此安装,并且在最新的驱动程序版本中一切正常。我该如何解决这个问题?

以下是我尝试运行的测试程序

import tensorflow as tf

# Creates a graph.
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
c = tf.matmul(a, b)
# Creates a session with log_device_placement set to True.
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
# Runs the op.
print(sess.run(c))

0 个答案:

没有答案