Ubuntu 升级 tensorflow1 为 2

2020-03-27  本文已影响0人  无解后仰跳投

在python3 环境下升级tensorflow-gpu 1.x 为 tensorflow2, 指定升级为2.0.0beta1版本。

pip3 install --user --upgrade tensorflow-gpu==2.0.0b1

同时可以安装tensorflow_probability这个概率统计库,安装命令如下

pip3 install --user tensorflow-probability==0.8

这里把版本设置为0.8,因为最新的版本0.9 (截止目前) 要适配tf2.1。

上一篇 下一篇

猜你喜欢

热点阅读