我有一个TensorFlow程序,旨在帮助我非正式地对我正在使用的GPU进行基准测试,更重要的是,理解TensorFlow附带的分析工具。代码只为一个矩阵设置两个占位符和一个matmul操作,然后调用一个会话来填充占位符并执行10次图。
以下是代码:
n = 10240
iter = 10
tf.reset_default_graph()
graph = tf.Graph()
with graph.as_default():
with tf.device("/gpu:0"):
matrix1 = tf.placeholder(tf.float32, [