tensorflow代码自动升级

来源:互联网 发布:光伏模拟软件 编辑:程序博客网 时间:2024/06/03 07:48

tensorflow升级:gpu版本的:

sudo pip install --upgrade tensorflow-gpu 

cpu版本的:

sudo pip install --upgrade tensorflow

升级遇到问题参考博客:http://blog.csdn.net/xyj1536214199/article/details/70766580
代码更新:
tensorflow0.12版本的代码升级到tensorflow1.0以上版本的代码有如下方法。
有两种方法一种是自动升级,调用函数:tf_upgrade.py,第二种是手动更改里面变动的函数。下面主要介绍自动升级,手动升级前提对两个版本的参数比较熟悉前提下可以进行。
tf_upgrade.py 使用说明
单个文件

对于单个文件,使用

python tf_upgrade.py --infile InputFile --outfile OutputFile

其中 InputFile 和 OutputFile 分别为你的旧代码和新代码,根据你的文件名做相应的替换,例如我的旧代码是test.py,将要生成的新代码是test_1.0.py,则

python tf_upgrade.py --infile test.py --outfile test_1.0.py

同时,tf_upgrade.py 会生成一个名为 report.txt 的文件,该文件记录了对旧文件做的所有改动,同时也给出了可能需要你手动更改的建议。

文件夹(目录)
和单个文件类似,使用

python tf_upgrade.py --intree InputDir --outtree OutputDir

例如,我的 programs 位于 /home/user/cool 目录,新的生成的文件我想放到/home/user/cool_1.0 里,则

python tf_upgrade.py --intree /home/user/cool --outtree /home/user/cool_1.0

参考地址:
http://blog.csdn.net/xyj1536214199/article/details/70766580
http://blog.csdn.net/u010099080/article/details/55260055

原创粉丝点击