import tensorflow as tf
print(tf.__version__)
print(tf.test.is_gpu_available()) # 结果True则是GPU环境
tensorflow检查是否是GPU环境
最新推荐文章于 2024-07-16 22:23:41 发布
import tensorflow as tf
print(tf.__version__)
print(tf.test.is_gpu_available()) # 结果True则是GPU环境