prlab 블로그

윈도우 10 환경에서 tensorflow gpu 설치하기(1)!! 본문

스터디흔적/인공지능 및 기계학습

윈도우 10 환경에서 tensorflow gpu 설치하기(1)!!

거녀엉 2018. 2. 22. 16:42

윈도우 10 환경에서 텐서플로우 GPU 버전 설치 가이드


설치 가이드 과정


1. cuda_8.0.61_win10 설치


2. cudnn-8.0-windows10-x64-v5.1 설치


3. Anaconda 설치


4. 가상환경 생성 및 tensorflow-gpu 설치


5.  Jupyter notebook에 가상환경 커널 추가하기!



제가 처음 GPU버전 텐서플로우를 설치하면서 구글링한 내용을 바탕으로 작성한 글입니다. 

저와 다른 환경에서는 이 설치가이드가 부적절할 수도 있으니 참고해주세요!

저의 설치 환경인 윈도우 10에서 설치파일 Anaconda3 4.2.0 버전, cuda_8.0.61_win10 버전, cudnn-8.0-windows10-x64-v5.1 버전으로 설치 및 실행 확인했습니다!


1. cuda_8.0.61_win10 설치

먼저 이 링크(https://developer.nvidia.com/cuda-downloads)로 NVIDA 홈페이지 접속 후 해당 버전의 exe파일 다운받습니다


아래 그림처럼 Legacy Release를 통해 CUDA Toolkit Archive에 들어가서 최신 버전이 아닌 이전 버전의 exe파일을 다운 받을 수 있습니다


Archived Releases에서 CUDA Toolkit 8.0 GA2 (Feb 2017) 클릭하면 해당 버전을 다운 받을 수 있음!






다운 받은 후 설치는 확인 및 다음만 눌러서 해주시면 됩니다(Setup 창 기본 설정 변경 없이 계속 설치 진행하세요!)


문제없이 CUDA 툴킷이 기본경로에 설치되었으면 "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin"경로가 생성되었을 것입니다. 그러면 이 경로를 제어판 -> 시스템 및 보안 -> 시스템-> 고급 시스템 설정 ->환경변수 안 시스템 변수 path에 추가시켜주세요!



환경 변수 추가는 bin폴더에 설치될 동적 라이브러리 파일(dll 파일)이 텐서플로우 GPU버전 설치시 필요하기 때문에 dll 파일 경로를 시스템변수 path에 추가해줍니다. 이것으로 툴킷 설치는 끝났습니다!



2. cudnn-8.0-windows10-x64-v5.0 설치

이 링크(https://developer.nvidia.com/cudnn)로 가서 cudnn-8.0-windows10-x64-v5.1 파일을 다운받아야 합니다!


1. Nvidia 사이트에 회원가입후 cuda 8.0 버전에 맞는 라이브러리(cudnn-8.0-windows10-x64-v5.1) 다운





Download cuDNN v5.1 (Jan 20, 2017), for CUDA 8.0 에서 cuDNN v5.1 Library for Windows 10로 들어가면 해당 버전을 다운 받을 수 있습니다




다운 받은후 cudnn-8.0-windows10-x64-v5.1.zip을 압축해제 하면 cuda 폴더가 생성되고 폴더안에 아래 그림처럼 bin, include, lib폴더가 있는 것을 볼 수 있습니다. 이 폴더를 "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0" 경로에 복붙해주면 다운받은 dll파일을 따로 환경변수 지정하지 않고 설치할 수 있어 간편합니다!




이렇게 복붙해서 덮어쓰기하면 cudnn-8.0-windows10-x64-v5.1 설치 가 완료되었습니다!!!


3. Anaconda 설치


아나콘다는 파이썬의 여러 라이브러리를 파이썬과 함께 설치해주는 배포 패키지(??)라고 할 수 있어서 파이썬 설치시 아나콘다로 설치하는 것이 매우 편리하고 가상환경 생성등 간편한 기능이 많음으로 아나콘다를 통해 파이썬을 설치했습니다.


이 링크(https://repo.continuum.io/archive/)에서 Anaconda3-4.2.0-Windows-x86_64.exe 파일을 다운 받아 주세요!!



설치파일을 실행해서 설치를 진행합니다. 계속 Next와 I agree 버튼을 눌러서 설치 진행하다 아래 창이 나오면 두 개의 체크박스 모두 체크해주세요 첫번째는 아나콘다를 환경변수 path에 등록하는것으로 윈도우 cmd창에서 별도의 아나콘다 접속없이 파이썬과 주피터 노트북을 편하게 실행할 수 있습니다!


두번째 체크박스는 아나콘다의 파이썬환경을 default 파이썬 환경으로 설정하는 것인데 그냥 체크해줍니다!



설치가 완료되면 cmd창을 켜서 설치가 잘 진행되었는지 확인합니다.


아래와 같이 명령어를 주게되면 파이썬 버전 및 conda, pip 버전도 확인 가능합니다.


C:\Users\PRLAB>python --version


C:\Users\PRLAB>conda --version


C:\Users\PRLAB>pip --version


C:\Users\PRLAB>python



위 창처럼 결과가 잘 출력된다면 아나콘다가 잘 설치된 것입니다.

0 Comments
댓글쓰기 폼