내용

글번호 1244
작성자 허진경 (나자바바)
작성일 2023-10-17 07:42:57
제목 TF 2.11부터 Windows에서는 CUDA 빌드가 지원되지 않습니다.
내용 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