Handling Devices in TensorFlow 2.0 (Colab)

Using CPU:

Using GPU:

Colab으로 GPU를 이용할 때에는 추가적인 설정이 필요하다.

1. 런타임 ➡ 런타임 유형 변경 선택
2. 하드웨어 가속기가 None으로 되어 있으면, GPU로 변경
   
3. 세션이 재시작되었으니, tensorflow를 다시 import를 해준다.
4. 아래와 같이 gpu_device_name을 얻어오는 것을 볼 수 있다.
   

또한, GPU를 사용할 시 GPU 지원 tensorflow library가 설치 되어 있어야 한다.
(아래는 notebook 전용 command)

!pip install tensorflow-gpu==2.3.0

위 조건이 만족된 상태에서 아래와 같이 GPU를 이용하여 tensorflow library를 사용할 수 있다.

 

+) 추가 - 내가 참고한 예제에서는 tf.test.is_gpu_available()을 사용했었는데 아래와 같은 deprecated warning이 떠서 tf.config.list_physical_devices()로 변경해 주었다. （￣︶￣）↗　

WARNING:tensorflow:From <ipython-input-5-1667dce961ac>:1: is_gpu_available (from tensorflow.python.framework.test_util) is deprecated and will be removed in a future version.
Instructions for updating:
Use `tf.config.list_physical_devices('GPU')` instead.​