tensorflow-gpu测试代码-python

tensorflow-gpu测试代码 首先导入TensorFlow库； 使用tf.Session()创建一个会话对象，并启动TensorFlow运行时环境； 定义两个输入占位符（placeholder），分别表示待执行矩阵乘法操作的两个矩阵； 定义矩阵乘法操作，使用tf.matmul()函数进行计算； 输入矩阵数据，使用feed_dict参数将数据传递给占位符； 使用sess.run()方法执行矩阵乘法操作，得到计算结果； 使用print()输出计算结果； 最后调用sess.close()方法关闭会话。 如果你的电脑上有可用的GPU，并且已经成功安装了CUDA和cuDNN等GPU加速库，那么TensorFlow-GPU将自动使用GPU来进行计算，从而提高计算效率。