tensorflow - 如何 : Import TensorFlow in Jupyter Notebook from Conda with GPU support?

Question

我已经使用 tensorflow website 中提到的 anaconda 环境安装了 tensorflow。完成后我的python安装路径发生了变化。

dennis@dennis-HP:~$ which python                                                                                                   
/home/dennis/anaconda2/bin/python  

并且安装了 Jupyter。我假设如果我能够在 conda 环境中导入和使用 tensorflow，我将能够在 Jupyter 中做同样的事情。但事实并非如此——

在我的系统中导入 tensorflow(不激活环境)

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 安装。

Answer 1

你有没有在 tensorflow 环境中安装过 jupyter？

类型 which jupyter找出答案。结果:

(tensorflow) [..]$ <anaconda_home>/envs/tensorflow/bin/jupyter # installed within the tensorflow environment.
(tensorflow) [..]$ <anaconda_home>/bin/jupyter                 # not installed.

如果未安装，请键入 pip install jupyter在 tensorflow 环境中。那就试试 import tensorflow再次在笔记本中。

希望这能有所帮助。