Tensorflow: thực hiện một ops với lõi cụ thể của CPU

Question

Hiện tại, bạn có thể chỉ định CPU hoặc GPU nào sẽ sử dụng với hàm tf.device (...) cho các ops cụ thể, nhưng ở đó bạn có thể chỉ định ở đâu một lõi của CPU?

Answer 1

Hiện không có API để ghim các op vào một lõi cụ thể, mặc dù điều này sẽ tạo ra một số feature request tốt. Bạn có thể ước tính chức năng này bằng cách tạo nhiều thiết bị CPU, mỗi thiết bị có một luồng chỉ một luồng, nhưng điều này không đảm bảo duy trì địa phương của giải pháp lõi-pinning:

with tf.device("/cpu:4"): 
    # ... 

with tf.device("/cpu:7"): 
    # ... 

with tf.device("/cpu:0"): 
    # ... 

config = tf.ConfigProto(device_count={"CPU": 8}, 
         inter_op_parallelism_threads=1, 
         intra_op_parallelism_threads=1) 
sess = tf.Session(config=config)