modelarts
/modelarts_faq/modelarts_05_0067.html
如何在Notebook中使用TensorFlow 2.X?
更新时间:2020/12/14 GMT+08:00
目前Notebook实例已支持TensorFlow 2.1,针对新创建的Notebook实例,可直接选择“Multi-Engine 2.0 (python3)”工作环境创建实例,包含TensorFlow-2.1.0引擎。
针对使用其他工作环境(“Multi-Engine 1.0 (python3 推荐)”或“Multi-Engine 1.0 (python2)”)创建的Notebook实例,如需使用TensorFlow 2.X,可参见如下步骤实现。
操作步骤
TensorFlow 2.0可以在Notebook中自行安装和使用,具体的步骤如下:
- 创建一个Notebook实例,打开并进入Notebook,选中并进入tensorflow1.13.1的kernel。
- 进入kernel页面安装TensorFlow 2.0(GPU版本)。输入如下代码然后单击代码上方的“Run”。
1
!pip install tensorflow-gpu==2.0.0b1
图1 安装TensorFlow 2.0 - 重启Kernel使更新生效。图2 重启Kernel
- 重启Kernel后,输入如下代码检查安装结果。即可开始使用TensorFlow 2.0了。
运行如下代码,当回显为“True”时,表示安装的是GPU版本。
1
print(tf.test.is_gpu_available())
图3 验证安装版本
父主题: Notebook
