텐서플로가 gpu 사용하고 있는지 확인하는 여러 방법/tensorflow-gpu 설치

pip install tensorflow-gpu

 

해주면 가장 최신 버전이 깔린다.

 

그런 다음 버전확인과 gpu를 사용하고 있는지 확인해준다.

 

 

import tensorflow as tf

print(tf.__version__)


tf.config.list_physical_devices('GPU') #2.xx 부터는 gpu_available 대신 이렇게 하라고 한다

 

현재 텐서플로 버전은 2.2 이고 gpu를 사용하고 있는것이 확인된다.

---

import tensorflow as tf
tf.debugging.set_log_device_placement(True)

# 텐서 생성
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)

print(c)

https://www.tensorflow.org/guide/gpu?hl=ko

GPU 사용하기  |  TensorFlow Core

 

텐서플로 튜토리얼에 의하면 위 코드는 MatMul 연산이 GPU:0에서 수행되었다고 보여줄 것이라고 한다. 

 

 

어떤 gpu를 사용중인지 궁금할 때는 이렇게 하면

 

 

from tensorflow.python.client import device_lib
device_lib.list_local_devices()

 

 

GeForce RTX 2080 Ti를 쓰고 있는것으로 확인 된다.

 

그리고 상시 모니터링을 하고 싶다면 터미널에서 watch -d -n 0.5 nvidia-smi 으로 

 

 

 

다음과 같이 gpu 메모리 사용량을 확인할 수 있다.