멀티 GPU 시스템에서 하나의 GPU만 사용하기

방법 1 tf.debugging.set_log_device_placement(True) try: # 유효하지 않은 GPU 장치를 명시 with tf.device('/device:GPU:2'): a = tf.constant([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]]) b = tf.constant([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]]) c = tf.matmul(a, b) except RuntimeError as e: print(e) ref: https://www.tensorflow.org/guide/gpu?hl=ko#%EB%A9%80%ED%8B%B0_gpu_%EC%8B%9C%EC%8A%A4%ED%85%9C%EC%97%90%EC%84%9C_%ED%95%98%EB..

2022.02.16