Tensorflow升级1.0版本

时间:2021-06-14 20:32:58

我的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