无法在jupyter-notebook中的GPU上运行Tensorflow

时间:2017-11-28 21:45:40

标签: tensorflow gpu jupyter-notebook

当我从jupyter笔记本运行时,我无法使用GPU运行tensorflow代码。如果我在python脚本中运行,相同的代码运行没问题。

我按照主要安装链接:

https://www.tensorflow.org/install/install_windows

也尝试过:

http://bailiwick.io/2017/11/05/tensorflow-gpu-windows-and-jupyter/

当我在python脚本文件中运行时,笔记本外没有问题。

问题很可能与此类似:

Tensorflow not running on GPU in jupyter notebook

更具体地说,我的测试: I can see both devices CPU and GPU via python a script

I can see only CPU via notebook

非常感谢您提前获得任何帮助!

2 个答案:

答案 0 :(得分:0)

很晚,但答案很简短:

Here是有关如何使用Docker设置基于GPU的Jupyterlab实例的教程(这使安装速度更快)。

我希望这会有所帮助!

答案 1 :(得分:-1)

我删除了所有现有环境并创建了一个新环境,解决了这个问题。

(另外,我必须应用以下内容来解决由移除的环境引起的问题:

https://github.com/jupyter/notebook/issues/2301