tensorflow cpu问题

返回:

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'

猜你喜欢

转载自www.cnblogs.com/wanghui-garcia/p/10912457.html