tf.config.set_visible_devices()

반응형

tf.config.set_visible_devices()이란?

tf.config.set_visible_devices(
    devices, device_type=None
)

device는 실제 GPU 장비의 이름이 들어가고, device_type에는 CPU 또는 GPU를 넣습니다.

 

만약 CPU만 사용할려면 device에 빈 리스트를 넣고 device_type에는 GPU를 넣으면 됩니다

예시) tf.config.set_visible_devices([], 'GPU')