내용 |
TF 2.11부터 Windows에서는 CUDA 빌드가 지원되지 않습니다. Windows에서 TensorFlow GPU를 사용하려면 WSL2에서 TensorFlow를 빌드/설치하거나 TensorFlow-DirectML-Plugin과 함께 tensorflow-cpu를 사용해야 합니다.
https://www.tensorflow.org/install/pip?hl=ko#windows-wsl2
python3 -m pip install tensorflow[and-cuda]
# Verify the installation:
python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"
----------
1. NVidia 그래픽 드라이버 설치
- https://www.nvidia.com/cuda-downloads
2. CUDA Toolkit 설치
- https://developer.nvidia.com/cuda-downloads
3. cuDNN 설치
- https://developer.nvidia.com/rdp/cudnn-archive
- zip을 내려받아 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2 에 압축 풀기
4. tensorflow-gpu 설치(관리자 권한으로 실행)
- pip install tensorflow-gpu
5. 확인은 아래 코드로(교재 242페이지 참고)
import tensorflow as tf
gpus = tf.config.experimental.list_physical_devices('GPU')
print(f'GPUs {gpus}')
if len(gpus) > 0:
try:
tf.config.experimental.set_memory_growth(gpus[0], True)
except RuntimeError:
pass |