tensorflow 1.8 测试gpu代码

最新推荐文章于 2023-07-26 23:17:26 发布

原创

最新推荐文章于 2023-07-26 23:17:26 发布 · 3.2k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#Tensorflow #GPU

查看日志信息若包含gpu信息，就是使用了gpu。

下面的代码查看日志信息，如果包含gpu信息就是使用了gpu

import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

输出

2018-08-30 13:06:50.335491: I tensorflow/core/platform/cpu_feature_guard.cc:140] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA
2018-08-30 13:06:50.431993: I tensorflow/stream_executor/cuda/cuda_gpu_executor.cc:898] successful NUMA node read from SysFS had negative value (-1), but there must be at least one NUMA node, so returning NUMA node zero
2018-08-30 13:06:50.432381: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1356] Found device 0 with properties: 
name: GeForce GTX 1080 Ti major: 6 minor: 1 memoryClockRate(GHz): 1.645
pciBusID: 0000:01:00.0
totalMemory: 10.91GiB freeMemory: 10.45GiB
2018-08-30 13:06:50.432412: I tensorflow/core/common_runti