tensorflow gpu 사용

Windows에서

>>> tf.config.list_physical_devices('CPU')
>>> tf.config.list_physical_devices('GPU')
>>> tf.config.list_physical_devices()	# 둘 다

로 각 장치를 확인할 수 있다.

GPU가 있고 cuda도 설치 했는데 tensorflow에서만 인식을 못 한다면
tensorflow 2.10 이하 버전을 설치해야 한다.

pip install tensorflow<2.11

이후 버전을 이용해야 한다면

• WSL2에 TensorFlow를 설치하거나
• tensorflow 또는 tensorflow-cpu 설치하고 선택적으로 TensorFlow-DirectML-Plugin을 사용
  해봐야 한다.

출처