jupyter notebook tensorflow打印device信息实例
juypternotebook中直接使用log_device_placement=True打印不出来device信息
#Createsagraph. withtf.device('/device:CPU:0'): a=tf.constant([1.0,2.0,3.0,4.0,5.0,6.0],shape=[2,3],name='a') b=tf.constant([1.0,2.0,3.0,4.0,5.0,6.0],shape=[3,2],name='b') c=tf.matmul(a,b) #Createsasessionwithlog_device_placementsettoTrue. sess=tf.Session(config=tf.ConfigProto(log_device_placement=True,allow_soft_placement=True)) #Runstheop. print(sess.run(c))
需要使用output_partition_graphs来输出device信息
#Createsagraph. withtf.device('/device:GPU:0'): a=tf.constant([1.0,2.0,3.0,4.0,5.0,6.0],shape=[2,3],name='a') b=tf.constant([1.0,2.0,3.0,4.0,5.0,6.0],shape=[3,2],name='b') c=tf.matmul(a,b) #Createsasessionwithlog_device_placementsettoTrue. sess=tf.Session(config=tf.ConfigProto(log_device_placement=True,allow_soft_placement=True)) #Runstheop. options=tf.RunOptions(output_partition_graphs=True) metadata=tf.RunMetadata() c_val=sess.run(c,options=options,run_metadata=metadata) printmetadata.partition_graphs
补充知识:Jupyter无法在控制台打印
因为数据有中文,所以我特意在jupter前面设置了
reload(sys)
sys.setdefaultencoding("utf-8")
结果使用print语句的时候无法输入内容。究其原因,是因为reload的时候把sdout变为ipython的对象了,所以要临时储存一下stdout的对象。
不妨试一试以下代码
importsys stdo=sys.stdout reload(sys) sys.setdefaultencoding('utf-8') sys.stdout=stdo
以上这篇jupyternotebooktensorflow打印device信息实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持毛票票。