Ubuntu 升级 tensorflow1 为 2

在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。

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容