文档首页 > > 常见问题> 开发环境> Notebook> 如何在Notebook中使用TensorFlow 2.X?

如何在Notebook中使用TensorFlow 2.X?

分享
更新时间: 2020/06/22 GMT+08:00

目前Notebook实例已支持TensorFlow 2.1,针对新创建的Notebook实例,可直接选择“TF-2.1.0-python3.6”工作环境创建实例。

如果需要在之前已创建的Notebook实例中使用TensorFlow 2.X,可参见如下步骤实现。

操作步骤

TensorFlow 2.0可以在Notebook中自行安装和使用,具体的步骤如下:

  1. 创建一个Notebook实例,打开并进入Notebook,选中并进入tensorflow1.13.1的kernel。
  2. 进入kernel页面安装TensorFlow 2.0(GPU版本)。输入如下代码然后单击代码上方的“Run”
    1
    !pip install tensorflow-gpu==2.0.0b1
    
    图1 安装TensorFlow 2.0
  3. 重启Kernel使更新生效。
    图2 重启Kernel
  4. 重启Kernel后,输入如下代码检查安装结果。即可开始使用TensorFlow 2.0了。
    运行如下代码,当回显为 “2.0.0-beta1”时,表示安装版本为TensorFlow 2.0。
    1
    2
    import tensorflow as tf
    print(tf.__version__)
    

    运行如下代码,当回显为“True”时,表示安装的是GPU版本。

    1
    print(tf.test.is_gpu_available())
    
    图3 验证安装版本
分享:

文档是否有解决您的问题?

提交成功!

非常感谢您的反馈,我们会继续努力做到更好!

反馈提交失败,请稍后再试!

*必选

请至少选择或填写一项反馈信息

字符长度不能超过200

提交反馈 取消

如您有其它疑问,您也可以通过华为云社区问答频道来与我们联系探讨

智能客服提问云社区提问