Windows 10 에 tensorflow-gpu 설치
1. 성공하는 설치환경 조합
: Anaconda는 사용하지 않음. Python3.5를 지원하는 Anaconda는 CUDA8.0까지만 지원하는것 같음. (가능한 사람은 설정건드려서 해보기 바람)
1) OS : Windows 10
2) Python3.5 (numpy는 1.14.3 사용)
3) CUDA 9.0
4) cuDNN v7.0
5) Tensorflow 1.6
- CPU로 돌리고 싶으면 설치시 pip install tensorflow 로 GPU로 돌리고 싶으면 pip install tensorflow-gpu 로 설치하면 됨
- tensorflow 1.8로 설치해보니 계속 오류나고 그래서 1.6으로 다시 설치함.
6) 내 노트북은 Nvidia 그래픽카드를 MX150을 사용
2. 설치상세
1) Python3.5 설치
- 아래 링크를 클릭하여 Python 3.5 버전 적당한거 받아서 설치한다. (여기서는 D:\ML_Program\Python35 에 설치함)
링크 : https://www.python.org/downloads/
2) Python의 path를 잡아준다.
- 내PC아이콘을 오른쪽 클릭해서 설정을 클릭한다. ==> 왼쪽에 "고급시스템 설정" 클릭 ==> 고급탭 클릭 후 하단의 환경변수 클릭
==> 아래 시스템변수에 보면 PATH 라는게 있다. (더블클릭, 없으면 새로생성해서 PATH를 만들어라)
- 아래의 3개를 추가한다.
D:\ML_Program\Python35 ==> 여기에 python.exe 가 있음.
D:\ML_Program\Python35\Scripts ==> 여기에 pip 등 여러 python 실행파일들이 있음.
D:\ML_Program\Python35\Lib\venv\scripts\nt ==> 여기에 tensorflow 환경을 activate, deactivate 하는 실행파일이 있음.
3) CUDA9.0 설치
- 링크를 클릭해서 CUDA 9.0을 다운로드 후 설치한다. [다운로드링크]
(Base 파일과 패치파일이 있는데 모두 다운로드해서 순서대로 설치하면 된다.)
- 설치시 CUDA 의 Path는 자동으로 셋팅된다.
4) CuDNN 설치
: CuDNN은 다운로드 후 위에 CUDA설치(CUDA Toolkit 설치폴더)한 위치에 동일한 구조로 덮어쓰기로 복사하면 된다.
CuDNN을 다운로드 받기 위해서는 먼저 회원가입을 해야한다. 회원가입 후 아래 다운로드를 클릭하자.
[다운로드 링크]
5) Tensorflow 1.6 설치
- windows의 cmd.exe를 연다.
- 다음 명령어로 설치한다.
pip install tensorflow-gpu==1.6 (CPU버전은 pip install tensorflow 라고 입력하면 됨)
6) Tensorflow가 잘 작동하는지 간단히 테스트해본다.
- windows의 cmd.exe 실행
- python 실행 : 콘솔창에서 python 을 치면 python 프로그래밍 입력으로 전환된다.
- 아래 명령어 입력 (tensorflow에서 hello라는 string을 출력하는 간단한 프로그램)
import tensorflow as tf
hello=tf.constant('hello')
sess=tf.Session()
print(sess.run(hello))
결과 ==> b'hello'
이렇게 하면 Tensorflow GPU는 설치완료
만족하셨나요? ~~~~~~~
#CUDA #텐서플로우 #cudnn #tensorflow-gpu #tensorflow설치 #windows10 tensorflow
