TensorFlow Lite移动端深度学习
上QQ阅读APP看书,第一时间看更新

2.3.2 使用Colaboratory开发并调试运行TensorFlow程序

Google在推出TensorFlow开源库之后,为了提高开发者的效率,特意推出了开发工具Colaboratory来协助开发者快速实现AI开发。Colaboratory是基于云端搭建的Jupyter Notebook环境,最大的好处是不需要进行任何配置就可以使用,并且完全在云端运行,开发者只需谷歌浏览器就可以开发并运行TensorFlow程序。

Jupyter Notebook是以网页的形式打开的,可以在网页中直接编写代码和运行代码,代码的运行结果也会直接在代码块下显示。如果在编程过程中需要编写说明文档,可在同一个页面中直接编写,以便做及时的说明和解释。

使用Colaboratory的好处如下。

· 可以在云端服务器中使用Jupyter Notebook创建Python程序文件。

· 可以在云端服务器中编写Python代码和TensorFlow代码。

· 可以在线运行Python程序和TensorFlow程序。

· 所有TensorFlow程序的编写和调试运行等工作都是通过浏览器实现的,无须开发者安装TensorFlow库,省略了搭建开发环境的工作,大大提高了开发效率。

具体步骤如下。

1)通过Google Chrome浏览器登录Colaboratory云端服务器,输入谷歌账号信息登录Colaboratory。依次单击“文件”→“新建笔记本”命令创建一个新的Jupyter Notebook文件,第一个文件会被自动命名为“Untitled0.ipynb”,如图2-7所示。

2)在弹出的界面中输入获取当前安装的TensorFlow的版本和路径的代码,如图2-8所示。

· 图2-7 创建一个新的Jupyter Notebook文件

· 图2-8 输入代码

上述代码的功能是,分别打印输出在当前计算机中安装的TensorFlow的版本和路径。因为这是在云端编写的代码,所以执行后会显示在云端服务器中安装的TensorFlow的版本和路径。单击按钮运行这段代码,执行效果如图2-9所示。

· 图2-9 在Colaboratory云端的执行效果

3)我们可以修改步骤1)中创建的Jupyter Notebook文件,假如想将“Untitled0.ipynb”的文件名修改为“first.py”,可以依次单击“文件”→“重命名笔记本”命令,如图2-10所示。

4)此时文件名“Untitled0.ipynb”将变为可编辑状态,将其重命名为“first.py”,如图2-11所示。

· 图2-10 单击“重命名笔记本”命令

· 图2-11 重命名为“first.py”

5)在使用Colaboratory时,可以设置GPU/TPU加速,方法是依次单击“代码执行程序”→“更改运行时类型”命令,如图2-12所示。

6)在弹出的对话框中可以选择硬件加速器,例如选择“GPU”,然后单击“保存”按钮,如图2-13所示。此时在Colaboratory中运行TensorFlow程序时,将使用云服务器提供的GPU加速器运行该程序,会发现运行速度会大大提高。

· 图2-12 单击“更改运行时类型”

· 图2-13 选择硬件加速器