#참고1. TensorFlow-GPU 사용

TensorFlow 사용 시, GPU 적용 방안.

GPU연결 상세
텐서플로우 - gpu

GPU 연결하는 코드

import os
import tensorflow as tf 
from tensorflow.python.client import device_lib
os.environ["CUDA_VISIBLE_DEVICES"] = "0"

GPU 연결확인

import tensorflow as tf 
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices() )

• GPU 사용

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"

• 다시 CPU 사용

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"