您的位置:首页 > 其它

Tensorflow升级1.0版本

2017-05-20 21:10 281 查看
我的tensorflow是从0.11版本升级到1.0版本的,基于python2.7的

升级方法

1、首先激活tensorflow环境

source activate tensorflow


2、输入以下安装命令:

GPU版本:

pip install --upgrade tensorflow-gpu


CPU版本:

pip install --upgrade tensorflow


输入完回车就开始升级过程,会下载并卸载重装一些必要的包,时间长短视网速而定。

3、使用脚本更新自己的旧代码:

  有两种方法更新旧的代码,一种就是用脚本自动升级,另一种就是手动更改。

使用脚本

  由于版本改动较大,TensorFlow 甚至出了一个 tf_upgrade.py 脚本来帮助你更新代码,从 github 获取这份代码,或者点击直接下载。

  对于更新单个文件,可以使用下面的命令:

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


例如,代码位于 /home/Tensor/CNN 目录,新的生成的文件放到 /home/Tensor/CNN_1.0 里,则

python tf_upgrade.py --intree /home/Tensor/CNN --outtree /home/Tensor/CNN_1.0


手动更改

  除了使用脚本自动更新外,还可以使用手动替换。

参考:http://blog.csdn.net/u010099080/article/details/55260055
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  tensorflow