diff --git a/tensorflow2fluid/README.md b/tensorflow2fluid/README.md index 20e12ceca8952dc851bec3341a74a56b0f76a689..d455fd231e77f02765a5b4941fff74a0af5b58b2 100644 --- a/tensorflow2fluid/README.md +++ b/tensorflow2fluid/README.md @@ -6,9 +6,9 @@ tensorflow2fluid支持将训练好的TensorFlow模型转换为PaddlePaddle模型 **我们计划专门梳理出指南文档,对比TensorFlow与PaddlePaddle的差异,帮助TensorFlow用户快速上手PaddlePaddle的使用,文档后续会整理在doc目录下,欢迎有需求的同学关注!** -## 依赖环境 +## 环境安装 -工具开发过程中,我们在如下环境配置中测试模型转换 +工具开发过程中,我们在如下环境配置中测试模型转换,建议使用anaconda > python == 2.7 or 3.6 @@ -16,7 +16,15 @@ tensorflow2fluid支持将训练好的TensorFlow模型转换为PaddlePaddle模型 > paddlepaddle == 1.3.0 - +``` shell +# pip install tensorflow-gpu +conda install tensorflow-gpu +pip install paddlepaddle-gpu + +# 上述安装过程可能会提示protobuf版本问题 +# 升级protobuf解决 +pip install protobuf --upgrade +``` ## 使用方法 本目录下提供了demo示例,展示如何将VGG_16模型转换为PaddlePaddle模型,详见[vgg_translate_tutorial](vgg_translate_tutorial.ipynb)