CPU版本的tensorflow代码改到GPU版运行时,会出现很多提示:
上面的提示可以通过加入下面的设置来屏蔽:
import os
os.environ['TF_CPP_MIN_LOG_LEVEL']='3'
最后的提示可以通过修改tensorflow import时的名字来消除:
import tensorflow.compat.v1 as tf
此时运行就是纯净的结果了。
另外做一些笔记:
1、 xxx.eval(),这些OP.eval()相当于是sess.run(OP),会执行这些以tf.开头的op。
2、GPU分布式训练的时候,在非chief_worker中屏蔽掉sv.summary_computed(),,否则运行失败。
3、启动顺序,先启动服务器ps,等待完成后,启动普通worker,最后启动chief_worker。也就是session的manager。
4、在CMD中cd到summary.filerwriter写入的路径下,运行
tensorboard --logdir C:\Users\xxx\.PyCharm2018.1\config\scratches\log\super
生成后,即可通过浏览器访问。这里注意,输入 http://127.0.0.1:6006/ 或者是使用CMD提示的地址是打不开的。使用下面的地址:
就可以看到tensorboard了