Ubuntu系统 Pycharm下启用 Jupyter Notebook时配置Tensorflow的要点

Pycharm是一款优秀的IDE工具,通过配置Anaconda及相关环境搭建(参见相关文章),可以在Pycharm中正确运行由Tensorflow开发的相关程序。
由于这些工具(NVIDIA驱动程序、CUDA、cuDNN、Tensorflow、Anaconda、Pycharm、Jupyter Notebook等)一直在持续地更新,在安装这些工具并学习Deep Learning相关应用的过程中,也要不停地适应各个软件版本变化之后的设置变化。
比如Tensorflow的版本更新到1.8以上后,已经可以支持CUDA9.0,相应的NVIDIA驱动也可以进一步更新,而tensorflow的网址也已经变更到tensorflow.google.cn
而Pycharm更新到2018.2这个版本后,相关的配置窗口与原先发生了较大的变化。
Pycharm中配置tensorflow相关环境的方法请参见本人博客“Pycharm下配置tensorflow的注意点
今天要讲的问题,不是在Pycharm中直接运行Tensorflow,而是通过Jupyter Notebook在浏览器中运行程序。
首先打开Terminal->输入jupyter notebook,这样浏览器就自动打开一个notebook页面
在Terminal中得到token
Ubuntu系统 Pycharm下启用 Jupyter Notebook时配置Tensorflow的要点
在Pycharm的项目中新建一个*.ipynb文件,将运行代码复制进这个文件中,点击项目上的绿色小箭头,运行,软件会弹出一个要求输入token的对话框。将之前在Terminal中得到的token复制到这个对话框中。
然后——这样未必能用,很可能报错为 no module named tensorflow
造成这一问题的原因是,*.ipynb文件的环境没有和py文件一样进行配置。(注意在之前的Pycharm版本中这两个文件在同一项目下似乎是同时配置的,而现在需要进行分别配置)
Ubuntu系统 Pycharm下启用 Jupyter Notebook时配置Tensorflow的要点
在运行界面的右上角可以 看到切换按钮,选择其中的Edit Configurations
Ubuntu系统 Pycharm下启用 Jupyter Notebook时配置Tensorflow的要点
重点是环境变量的配置,Enviroment variables
以及工作路径 Working directory
这两项的设置应与在pycharm中运行时同样设置。
配置好以后再在浏览器中运行cells
Ubuntu系统 Pycharm下启用 Jupyter Notebook时配置Tensorflow的要点
Ubuntu系统 Pycharm下启用 Jupyter Notebook时配置Tensorflow的要点
注:图中所运行的代码来自机械工业出版社《深度学习之TensorFlow入门、原理与进阶实战》一书。