这个屌丝很懒，什么也没留下！

测试GPU是否参与模型训练等运算_跑模型时怎么判断gpu参与了没

作者：木道寻08 | 2024-08-05 15:34:10

跑模型时怎么判断gpu参与了没

import tensorflow as tf
# Creates a graph.
a = tf.constant([6.0, 5.0, 3.0, 3.0, 2.0, 1.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=1
2
3

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/木道寻08/article/detail/932969