【发布时间】:2016-11-09 02:16:27
【问题描述】:
我已经使用 tensorflow website 中提到的 anaconda 环境安装了 tensorflow,并且在更改了我的 python 安装路径之后。
dennis@dennis-HP:~$ which python
/home/dennis/anaconda2/bin/python
并且安装了 Jupyter。我假设如果我能够在 conda 环境中导入和使用 tensorflow,我将能够在 Jupyter 中做同样的事情。但事实并非如此——
在我的系统中导入张量流(不激活环境)
dennis@dennis-HP:~$ python
Python 2.7.11 |Anaconda 4.1.0 (64-bit)| (default, Jun 15 2016, 15:21:30)
[GCC 4.4.7 20120313 (Red Hat 4.4.7-1)] on linux2
Type "help", "copyright", "credits" or "license" for more information.
Anaconda is brought to you by Continuum Analytics.
Please check out: http://continuum.io/thanks and https://anaconda.org
>>> import tensorflow as tf
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
ImportError: No module named tensorflow
>>> exit()
在 conda 环境中导入 tensorflow
dennis@dennis-HP:~$ source activate tensorflow
prepending /home/dennis/anaconda2/envs/tensorflow/bin to PATH
(tensorflow) dennis@dennis-HP:~$ python
Python 2.7.12 |Continuum Analytics, Inc.| (default, Jul 2 2016, 17:42:40)
[GCC 4.4.7 20120313 (Red Hat 4.4.7-1)] on linux2
Type "help", "copyright", "credits" or "license" for more information.
Anaconda is brought to you by Continuum Analytics.
Please check out: http://continuum.io/thanks and https://anaconda.org
>>> import tensorflow as tf
I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcublas.so locally
I tensorflow/stream_executor/dso_loader.cc:102] Couldn't open CUDA library libcudnn.so. LD_LIBRARY_PATH: /usr/local/cuda-7.5/lib64
I tensorflow/stream_executor/cuda/cuda_dnn.cc:2092] Unable to load cuDNN DSO
I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcufft.so locally
I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcuda.so locally
I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcurand.so locally
由于上述导入成功,我尝试在 jupyter 中执行相同操作 (在环境中启动 jupyter) 但导入时出现以下错误 -
ImportError Traceback (most recent call last)
<ipython-input-1-41389fad42b5> in <module>()
----> 1 import tensorflow as tf
ImportError: No module named tensorflow
我的猜测是笔记本没有在 conda 环境中运行。那么,你能告诉我如何强制它做同样的事情吗?
或者你可以给我提供有关如何在 jupyter 中导入 tensorflow 的详细信息
编辑#1:
我已经使用conda install -c jjhelmus tensorflow=0.9.0 命令在anaconda 安装中成功安装了tensorflow。 [来源:conda.anaconda.org/jjhelmus]
但这会禁用 GPU 支持,因此类似下面的代码会返回错误
with tf.Session() as sess:
with tf.device("/gpu:0"): #GPUs are not enabled on the system so it throws an error
matrix1 = tf.constant([[3., 3.]])
matrix2 = tf.constant([[2.],[2.]])
product = tf.matmul(matrix1, matrix2)
result = sess.run([product])
print result
那么,如何启用 GPU 支持?是否有替代解决方案可以在支持 GPU 的 conda 中安装 tensorflow?
编辑 #2:
提到here,只有在为目标 GPU 构建源代码时才有可能支持 GPU。 如果是这样,请提供有关如何完成此操作的详细信息,以便我安装启用 GPU 的 tensorflow。
【问题讨论】:
-
你在哪个平台上?如果你在 linux 上,谷歌现在已经制作了一个预构建的二进制文件,你可以很容易地与 anaconda 一起使用。 tensorflow.org/versions/r0.9/get_started/…
标签: tensorflow anaconda jupyter nvidia conda