Cấu hình Tensorflow sử dụng tất cả của CPU

Question

Reading: https://www.tensorflow.org/versions/r0.10/resources/faq.html nó khẳng định:

Liệu TensorFlow tận dụng tất cả các thiết bị (GPU và CPU) có sẵn trên máy tính của tôi?

TensorFlow hỗ trợ nhiều GPU và CPU. Xem tài liệu hướng dẫn cách sử dụng về cách sử dụng GPU với TensorFlow để biết chi tiết về cách TensorFlow chỉ định hoạt động cho thiết bị và hướng dẫn CIFAR-10 cho mô hình ví dụ sử dụng nhiều GPU.

Lưu ý rằng TensorFlow chỉ sử dụng thiết bị GPU có khả năng tính toán lớn hơn 3.5.

Điều này có nghĩa là Tensorflow có thể tự động tận dụng tất cả CPU trên máy đã cho hoặc không được định cấu hình rõ ràng?

Answer 1

CPU được sử dụng thông qua "thiết bị" chỉ là một luồng. Bạn có thể kiểm soát số lượng chủ đề nếu bạn cảm thấy mình cần nhiều hơn:

sess = tf.Session(config=tf.ConfigProto(
    intra_op_parallelism_threads=NUM_THREADS))