window에서 tensorflow-gpu 사용하기

 

리눅스 os가 아닌 window10에서도 딥러닝(with tensorflow)을 돌릴 수 있도록 메뉴얼을 만들어보았다.

 

https://www.tensorflow.org/install/source_windows?hl=ko#tested_build_configurations

 

 

---

 

1. Python 설치

이미 3.10이 설치되어 있어서 생략함

https://www.python.org/downloads/

Download Python

 

 

2. CUDA와 cuDNN 설치

 

- CUDA

 

 

https://developer.nvidia.com/cuda-11.2.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

 

 

요즘 다 visual studio code 쓰니까 visual Studio를 설치하지 않고 CUDA Toolkit을 설치하려면, 체크박스를 선택한 상태로 Next 버튼을 클릭하여 설치를 계속 진행하기 (예전 visutal studio 20xx 내용이였음)

- cuDNN

https://developer.nvidia.com/rdp/cudnn-archive

cuDNN Archive

 

 cudnn-11.2-windows-x64-v8.1.1.33.zip이 깔리는데 풀고 추가 작업이 있다!

4. 설치 확인

 

 

 

CUDA와 cuDNN 환경 변수 설정

 

환경 변수 설정:

1. 시스템 속성 창을 열고 고급 시스템 설정으로 이동
2. 환경 변수 클릭
3. 시스템 변수 섹션에서 새로 만들기를 클릭하여 CUDA_PATH라는 새 변수를 추가하고, 값을 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2로 설정합니다.
4. 시스템 변수 섹션에서 Path 변수를 찾아 편집을 클릭하고 경로 추가

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\libnvvp
# C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\extras\CUPTI\lib64

 

 

보면 경로 두 개는 깔려 있따.

 

 

 

추가작업

 

 

cuDDN 압축 푼 파일들을 이동시켜야 하는데

1. bin >> bin 

2. include >> include 

3. lib >> extras\CUPTI\lib64

차례대로 이동 시킨 후 

 

만능의 *재부팅*

import tensorflow as tf
print("Num GPUs Available: ", len(tf.config.list_physical_devices('GPU')))

 

 

위와 같이 1이 뜨면 완료!!