Tensorflow gpu 버전 설치하기, windows 10, Anaconda(python 3.6)

안녕하세요 이번 포스팅에서는 아나콘다를 이용해서 윈도우 10환경에서 tensorflow gpu 버전을 설치해볼까 합니다. 

 

 텐서플로우의 경우는 python 버전에 따라 지원을 공식적으로 하기도 하고 안하기도 하기 때문에 정확하게 python 버전을 맞출 필요가 있습니다. 

그럼 시작하겠습니다. 

 

※ 요구사항

----------------------------------------------------------------------------------------------------------------------------------

NVIDIA GPU (GTX 1050ti이상이면 충분하다)

Anaconda with python 3.6

CUDA Toolkit (ver 9.0)

CUDNN (ver 7.0)

----------------------------------------------------------------------------------------------------------------------------------

 

먼저 CUDA Toolkit 은 아래 링크에서 다운로드 받을수 있습니다. 

Download CUDA Version 9.0 here: https://developer.nvidia.com/cuda-90-download-archive

 

그리고 CUDNN의 다운로드 링크는 아래와 같습니다. 

Download CUDNN Version 7.0 here: https://developer.nvidia.com/rdp/cudnn-download

 

Anaconda의 다운로드 링크는 아래와 같습니다. 

Download Anaconda here: https://www.anaconda.com/distribution/

Anaconda Python/R Distribution - Free Download

 

여기서 주의할 점은 Tensorflow의 경우 python 3.5, 3.6 을 지원합니다. 

따라서 Anaconda에 현재 깔려있는 python 버전을 체크하시면 3.7버전으로 되어있을 겁니다. 

참고로 Anaconda 실행 후 Python 이라고 치면 아랫줄에 현재 설치된 python 버전에 대한 정보를 얻을 수 있습니다. 

현재 Anaconda에 설치된 python 버전, 3.7.3

그리고 설치가능한 파이썬 목록은 다음과 같습니다. 

Conda serach python 명령어를 치면 설치 가능한 python 버전을 확인할 수 있습니다. 

conda search python 명령어 실행

 

여기서 두가지 선택지가 생깁니다. 

python 3.7버전을 냅두고 pyhton 3.6 가상환경을 설치할 것인지 아니면 아예 3.6버전을 새로 설치할지 의 선택지입니다. 

 

사실상 가상환경으로 설정하는 편이 가장 좋은 방법입니다. 

기존 버전을 지울 필요 없이 가상환경만 activate하면 되기 때문입니다. 

anaconda에 설치된 가상환경 목록 확인

먼저 위의 그림과 같이 conda info --envs라는 명령어를 치면 가상환경 목록이 나옵니다. 

 

만약 목록 중 가상환경을 삭제하고 싶다면 아래의 명령어를 사용하면 됩니다. 

conda remove --name 가상환경이름 -all

 

그리고 나서 python 3.6버전의 tensorflow 가 실행 가능한 아나콘다 환경을 만들 차례다.

아래의 그림과 같이 conda create -n tensorflow python =3.6 anaconda 라고 타이핑 하자

conda -n tensorflow python=3.6 anaconda 명령어 실행

 

가상환경이 정상적으로 설치가 되었다면 다음과 같은 화면을 볼 수 있을 것이다.

그리고 나서 설치된 가상환경을 실행시켜보면 마찬가지로 아래 화면처럼 나올 것이다. 

정상적으로 설치가 되었으며, 가상환경 activate한 후 화면

 

이렇게 했다면 python 3.6버전이 실행되는것을 확인 가능하다. 

python 3.6버전 실행 확인 

 

인제 pip를 이용해서 tensorflow -gpu 및 필요 라이브러리들을 설치하면 됩니다. 

 

tensorflow-gpu 버전 pip를 이용해서 설치하는 화면

 

 

제 포스팅은 여기까지 입니다. 

포스팅을 봐주셔서 감사합니다. 

다음에도 좋은 포스팅으로 작성하겠습니다.