Windows 10에서 tensorflow 설치하기

feelpassion (25)in #tensorflow • 7 years ago

글을 쓰고 있는 중입니다 !! 여러개의 글을 한 꺼번에 쓸 수는 없네요^^;

설치 동기

사실 윈도우 운영체제에서 tensorflow를 지원하게 된 것은 얼마 되지 않았다. Ubuntu Linux에서 tensorflow를 써보긴 했는데, 윈도우 환경에서 된다니, 잘 돌아가는지 호기심도 생기고, 집에서 공부해보고 싶은 생각에 설치를 해보기로 했다.

그리고,
집에 PC가 한 대 밖에 없는데, 가족들과 같이 써야하는 상황이라 윈도우를 지우고 Ubuntu를 깔 수도 없고, 무엇보다 GPU가 VirtualBox에서 인식이 안된다니 VirtualBox에 Ubuntu를 깔아서 쓸 수도 없다. 이래저래 윈도우 운영체제를 선택할 수 밖에 없는 상황이다.

설치할 컴퓨터 사양은 아래와 같다.
Windows10, i7-7700, 16GB, 500GB SSD, 4TB HDD, GTX1080Ti

설치 순서

먼저 GPU가 있으면 GPU의 디바이스 드라이버를 설치한다. 아래의 링크에서 소유하고 있는 GPU에 맞는 드라이버를 검색해서 설치하자.
http://www.nvidia.co.kr/Download/index.aspx?lang=kr
다운로드 받은 파일을 실행하자.
라이센스 동의하고...
귀찮으니 빠른 설치로 선택
tensorflow의 install 페이지에서 Installing Tensorflow on Windows를 선택한다.
그러면, GPU를 지원하도록 설치할 것인지, CPU만 지원하도록 설치할 것인지 선택하라고 한다. GPU가 있으므로 GPU를 지원하는 방법으로 설치하려 한다. 먼저, GPU 구동을 위해 NVIDIA 사에서 제공하는 CUDA toolkit과 cuDNN library를 설치해보자.
꼭!! tensorflow 설치 페이지에 적혀있는 버전의 CUDA와 cuDNN을 설치해야 한다!!
만약, 최신 버전을 무작정 설치했다면, 그 버전을 지우고 다시 설치해야하는 불상사가 생긴다.

A. CUDA Toolkit download https://developer.nvidia.com/cuda-downloads
위 페이지에서 무심결에 최신 버전을 다운받으면 안된다! 9.x의 최신 버전이 아니라, 우리가 다운로드 받아야할 버전은 8.0이다. 그래서 페이지 아랫 쪽에 있는 Legacy Releases를 클릭한다.

B. 어라? CUDA 8.0이 GA1와 GA2 두 가지 버전이 있네! 그 동안의 경험으로 봤을 때, 같은 버전이 두 가지라는 것은 먼저 나온 것이 결함이나 보완할 점이 있는데 기능적으로 변화가 거의 없을 때인 경우가 많다. 두 버전의 Release Note를 보니 세부 버전의 번호가 다르며, GA2가 GA1의 버그를 고쳤다고 되어 있으니 예상이 맞았다. GA2 버전으로 다운받자.

C. Base Installer와 Patch 2가 있는데 순서대로 둘 다 설치하자. 32bit 윈도우 운영체제를 설치해서 쓰는 분들은 번거롭겠지만 64bit 윈도우를 재설치해야 이 후의 과정을 진행할 수 있다. 32bit 운영체제는 이제 설 자리가 없다ㅜㅜ

D. 설치 파일을 실행하면 설치를 위해 압축 파일을 풀 임시 디렉토리를 지정하라고 한다. 설치가 끝나면 저절로 사라지는 디렉토리이니 아무데나 지정해도 상관없다.