无法从 jupyterhub/jupyter notebook 调用 tensorflow gpu，为什么？

Question

好吧，我认为八个小时足够我自己解决这个问题，所以我只问大家:

我在 jupyterhub 和 Jupyter notebook 之外的名为“tensorflow”的虚拟环境中运行 tensorflow-gpu 1.1.0 就好了。也就是说，我可以导入 tensorflow 并使用 gpu 运行脚本。

当我在我的 tensorflow virtualenv 中并使用 jupyterhub 时，Jupyter 似乎无法“看到”tensorflow。我收到以下错误:

ImportError: libcublas.so.8.0: cannot open shared object file: No such file or directory

1) 这是一条常见的错误消息，指示 tensorflow 安装问题，但我的路径和环境变量似乎没问题。毕竟，我可以在 Jupyter 之外很好地使用 tensorflow-gpu。

2) 输入“which jupyter”显示 ~/anaconda3/envs/hub/bin/jupyter，所以我相信我在我的 virtualenv 中引用了 jupyter。3) Pip freeze 显示我有jupyterhub和tensorflow-gpu。我什至做了一个 pip3 freeze，它也显示了两个包。

有什么想法吗？ tensorflow-gpu 可以从 Jupyter 笔记本运行吗？

Answer 1

我从这里得到了解决方案:

[ https://github.com/jupyter/notebook/issues/1290][1]

基本上，jupyter 出现了“错误”，因为它无法读取我的 LD_LIBRARY_PATH 变量。我确实将所有内容都正确地放入了 .bashrc 中，所以我不知道为什么。

切换到命令行(终端)。切换到您的虚拟环境(如果有的话)。

输入:jupyter notebook --generate-config

它会告诉您 jupyter 配置文件的存储目录。如果你想再次列出它，请输入:jupyter --config-dir

我的 jupyter_notebook_config.py 文件位于:/home/me/.jupyter/jupyter_notebook_config.py

在此文件 jupyter_notebook_config.py 的最顶部，添加以下代码:

import os
c = get_config()
os.environ['LD_LIBRARY_PATH'] = '/usr/local/cuda-8.0/lib64:usr/local/cuda-8.0/lib64/libcudart.so.8.0'
c.Spawner.env.update('LD_LIBRARY_PATH')

然后重启jupyterhub或者jupyter notebook(在命令行输入:jupyter notebook

Tensorflow GPU 应该可以工作。

即使您正在运行 jupyterhub，同样的事情也适用。在 jupyter 中进行更改，而不是在 jupyterhub 中进行更改。 (jupyterhub 的每个用户都有自己的 jupyter 进程，因此不要在“hub”级别进行更改，而是在 jupyter notebook 级别进行更改。