это нормально для Keras не загружать много CPU & GPU?

Question

Мой P C не так хорош: i7-2600 и GeForce GTX 970. Но когда я запускаю скрипт, который классифицирует фотографии с CNN, моя система загружается не сильно: процессор на 20%, графический процессор на 5%.

Похоже, что доступен графический процессор:

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 4011293230370962537
, name: "/device:GPU:0"
device_type: "GPU"
memory_limit: 3110401638
locality {
  bus_id: 1
  links {
  }
}
incarnation: 3815751383857027347
physical_device_desc: "device: 0, name: GeForce GTX 970, pci bus id: 0000:01:00.0, compute capability: 5.2"
]

Моя версия Keras

print(keras.__version__)
2.3.1

Моя версия Tensorflow

print(tensorflow.__version__)
1.14.0

Answer 1

Вы можете установить долю GPU, которую хотите использовать, запустив этот фрагмент:

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=1) #number between 0 and 1
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))

По опыту, вы должны запустить его сразу после выполнения import tensorflow. Перезапустите ядро, если вам нужно.