返回:
2019-05-23 14:33:04.952328: I tensorflow/core/platform/cpu_feature_guard.cc:140] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA 2019-05-23 14:33:05.050837: 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 2019-05-23 14:33:05.051192: 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.683 pciBusID: 0000:01:00.0 totalMemory: 10.91GiB freeMemory: 10.60GiB 2019-05-23 14:33:05.051205: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1435] Adding visible gpu devices: 0 2019-05-23 14:33:05.225183: I tensorflow/core/common_runtime/gpu/gpu_device.cc:923] Device interconnect StreamExecutor with strength 1 edge matrix: 2019-05-23 14:33:05.225212: I tensorflow/core/common_runtime/gpu/gpu_device.cc:929] 0 2019-05-23 14:33:05.225218: I tensorflow/core/common_runtime/gpu/gpu_device.cc:942] 0: N 2019-05-23 14:33:05.225364: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1053] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 10263 MB memory) -> physical GPU (device: 0, name: GeForce GTX 1080 Ti, pci bus id: 0000:01:00.0, compute capability: 6.1)
解决:
在代码中添加:
import os os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'