개발 이야기/Python

tensorflow gpu 가속 기능 사용 설정

가끔.하늘 가온아 2021. 1. 12. 11:15
728x90
반응형

[개발 환경] windows 10 / PyCharm / Python 3.8 / CUDA Ver 10.1 / tensorflow, tensorflow-gpu 2.3.1

 

gpu 가속을 사용하려면 tensorflow-gpu, 그리고 cuda toolkit, cuda DNN을 설치해야 합니다.

 

최신 cuda 버전은 11.2 이지만 tensorflow 2.3.1에서는 CUDA Toolkit 10.1, cuDNN 7.x 버전을 사용합니다.

 

다운로드 주소는 각각 아래와 같습니다.

 

CUDA Toolkit 10.1

    developer.nvidia.com/cuda-toolkit-archive (2.3GB)

 

CUDA Deep Neural Network Library (cuDNN)  7.6.5

   developer.nvidia.com/rdp/cudnn-archive (264MB)

(* cuDNN 파일을 다운받기 위해서는 nvidia developer 사이트 가입이 필요합니다. 비용이 추가되지 않으니 안심하세요. ^^;)

 

CUDA Toolkit은 설치 파일을 제공되어 exe 파일을 실행해 바로 설치할 수 있습니다. 

 

cuDNN은 library 사용을 위한 파일만 제공됩니다. 압축 파일로 제공되며 압축을 풀면 아래와 같은 폴더가 나옵니다.

 

모든 폴더 파일을 "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1"에 복사하시면 추후 사용하기 편리합니다. 

 

python에서의 사용을 위해서는 필요한 dll 파일들을 c:\windows\system32에 복사해야 합니다.

 

"C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin" 폴더에서 "cu"로 시작하는 dll 파일 12개 (cudnn 파일 포함)를 복사하여 system32 폴더에 복사하면 됩니다.

 

샘플을 돌려보니 성공적으로 인식하네요.

 

반응형