[인공지능]anaconda로 tensorflow-gpu, cuda 사용하기 환경 설정

anaconda로 tensorflow-gpu 사용하기 환경 설정

---

필자가 도커로 매일매일을 오타와 자동완성 없이 싸워오다가 docker의 용량이 19gb나 되어버려서 c의 용량이 너무 부족했었다. 이미지가 너무 컸나보다.... 그래서 다른방법을 찾는 중 윈도우에서 아나콘다를 사용해서 텐서플로우를 사용하는 방법을 찾았다. 그래서 오늘은 윈도우에서 tensorflow-gpu를 사용해보고 pycharm에 적용 시켜보겠다.

1. anaconda3 설치

우선은 아나콘다 다운로드 사이트에 들어가서 아나콘다를 다운받는다.

2. cuda와 cudnn설치 

1. cuda 9.0 설치 

                                

https://developer.nvidia.com/cuda-toolkit-archive 사이트로 이동해서 우선 base를 다운 받은 후 설치한 후에 나머지 두개를 다운로드 후 설치해주면 된다. cuda9.0을 다운 받고 설치한다.

2. cudnn

https://developer.nvidia.com/rdp/cudnn-archive 사이트로 이동해서 cuda9.0버전의 7.0.5버전을 다운 받았다.

다운 받은 후에 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0 이곳에 복사 붙여넣기 해주면된다. 

3. 환경변수 설정해주기

제어판에서 시스템 및 보안 -> 시스템  -> 고급 시스템 설정 -> 고급 -> 환경변수에 들어간다. 

유저네임에 대한 사용자 변수에서 

Path에 편집을 눌러서 추가 해준다.

그 후에 시스템 변수

위와 같이 3개 넣어줬다. 

CUDA_HOME C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

CUDA_PATH C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

CUDA_PATH_V9_0 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

이렇게 넣어주었다. 

3. anaconda prompt에서 다운받기 

검색창에서 anaconda prompt를 실행시켜서 

 

> conda create -n tensorflow pip python=3.6

> activate tensorflow > (tensorflow)

> pip install --ignore-installed --upgrade tensorflow-gpu 

> python

python에 들어가서 import tensorflow를 쳤을 떄 아무 오류가 없으면 똑바로 설치된 것이다. 

설치한 다음부터 텐서플로우 코드를 실행시키면 

위와 같이 필자의 그래픽 카드의 정보가 나오기 시작했다. 

4. pycharm에 적용시키기

1. pycharm 제일위 네비게이션중 file에서 setting에 들어간다. 

2. 

3. project Interpreter에 들어가서 C:/Users/유저네임/Anaconda3/envs/tensorflow/python.exe를 추가해주면 pycharm에서도 사용가능하게 된다.