| 내용 |
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 |