윈도우7 환경에서의 파이참 Tensroflow-GPU 설치!!
프로젝트를 진행하는 과정에서 보드에서는 학습이 안 된다는 '한컴MDS' 사 직원 분의 소중한 조언을 참고해서 컴퓨터에서 학습시키고 학습된 모델을 보드에서 돌리는 것으로 방향을 설정하였습니다....(미리 알았더라면,...ㅠㅠ)
무튼 학습을 시키기 위해 일단 프로젝트에서 사용하는 nvidia 사의 Jetson TX2 보드에서 지원하는 머신 러닝 엔진을 찾아보니 다행히도 구글의 'TensorFlow'가 이제 지원이 된다고 나와 있다!!
TensorFlow 모델을 nvidia 사에서 개발한 TensorRT에 맞는 모델 형식 파일로 변환한다면 보드에서 최적으로 돌아갈 수 있다는 사실을 확인하였고, 그에 앞서 TensorFlow를 통한 학습을 진행하고자 했습니다.
(※참조. TensorFlow로 학습한 neural network 모델을 변환할 수 있는 방법에 대한 웹미나가 이번년도 초에 진행됐는데..... 일단 관련 메일인데.... 당시 우리나라 시간으로 새벽이라 못 참가했다....ㅠㅠ
이 당시에는 웹미나에 참가 못했는데.... 다행히!! NVIDIA에서 녹화 영상을 오픈해줘서 나중에 공부할 수 있게 해줬다! 감사합니다!!ㅠㅠ)
무튼 이제부터 'Window 7'에 'TensorFlow-GPU'를 까는 방법에 대해 포스팅하겠습니다~~~
먼저 OS를 확인하기 위해 '내컴퓨터 -> 속성'에 들어가서 OS 확인!
이제부터 깔아봅시다! 저는 GPU 버전을 깔기 위해서 '1. CUDA Toolkit 설치 - 2. CuDNN 설치 - 3. 파이썬 설치 - 4. TensorFlow 설치' 순으로 진행했습니다!
간단하게 이용하실 분들을 1,2 과정을 뛰어넘으시고 바로 3번부터 시작하면 CPU 버전으로 설치가 쉬울 거에요!
먼저 참고한 TensorFlow 홈페이지!
https://www.tensorflow.org/install/install_windows
솔직히 차근차근 읽어가면서 진행하면 바로 설치가 됩니다....ㅋㅋ 이 사이트를 참조해서 하는 게 제일 최신 설치법일게에요! 제 블로그는 참고해주시고 설치한 버전 꼭 지켜주세요!
1. CUDA Toolkit 9.0 설치 (GPU 버전만!)
https://developer.nvidia.com/cuda-90-download-archive
설치 방법은 간단! 해당 링크에 들어가서 내 컴퓨터의 OS와 Architecture, version을 선택하고 installer type을 추가로 선택한 뒤 설치하면 됩니다!
설치한 exe파일을 실행시키면 바로 설치 진행...
(컴퓨터마다 차이가 있겠지만 저는 10분 정도 소요됐네요...ㅎ)
2. CuDNN 7.0 설치!(GPU 버전만)
https://developer.nvidia.com/rdp/cudnn-archive
cuDNN Archive
cuDNN 설치를 위해서는 nvidia 홈페이지에서 로그인이 필요합니다!
그 후에 꼭 설치한 CUDA와 호환이 되는 cuDNN을 설치해주세요!
설치한 zip 파일을 CUDA가 설치된 폴더의 /bin 위치에 압축해제!(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0)
여기서 이제 PATH를 추가시켜야 됩니다!
'내컴퓨터 -> 속성 -> 고급 시스템 설정 -> 환경변수'에 들어가서
'Path' 또는 'PATH'를 찾아서 마지막 부분에 추가!!(꼭 마지막 부분에 추가입니다! 기존에 있던 거 삭제하시면 고생합니다...예전 저처럼...)
마지막 부분에 추가할 부분 ->cuDNN 압출 해제 폴더의 bin! 제 경우에는 'C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\cuda\bin' 입니다
그리고 확인!
3. python 3.6 설치!
https://www.python.org/downloads/
저는 python-3.6.4 버전을 설치했습니다!
4. TensorFlow GPU 설치
'cmd' 창을 관리자 권한으로 실행한 후 'pip3 install --upgrade tensorflow-gpu' 명령어 실행!
(* CPU버전 'pip3 install --upgrade tensorflow' 명령어 실행 )
5분 뒤 설치완료!!
확인을 위해 파이썬 설치할 때 깔았던 IDLE 실행!
그 후 명령어 창에
import tensorflow as tf
실행해서 오류가 생기는지 아닌지만 확인!
(오류 생겼다면.... 다시 처음부터....ㅠㅠ)
이런 식으로 커서가 넘어가면 파이썬에 tensorflow 모듈이 설치가 됨을 의미해요!
5. pycharm에서 실행
마지막으로 pycharm을 실행합니다!
먼저 새로운 프로젝트 생성
그 후에 'File - settings - project interpreter'를 누르시면 이 창이 뜹니다.
여기서 '+'창 클릭 후 입력 창에 'tensorflow-gpu'를 검색한 후 좌측 하단에 'install package'라는 버튼 클릭!!
그 후 파이참 하단 표시줄에 보면 package 설치 진행 상황이 나타나면서 설치 완료 후에는 'package installed successfully'라는 완료 메세지가 출력됩니다.
드디어 다 됐는지 확인을 위해 파이썬 파일을 생성
그 후 명령어 창에
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
기입한 후 'Run'을 실행하게 되면...!
결과 터미널 창에 출력 메세지가 나옵니다!!
드디어 설치 완료!!
그나마 윈도우 환경이 설치가 빠르네요...ㅋㅋㅋ 리눅스는 진짜 어휴.....멘붕이였는데...
이제 프로젝트를 위한 환경 구축 완료~~ 열심히 해보자!
아! 만약 텐서플로우 실습을 해보고 싶은 분들에게는 '김성훈 교수님'의 '모두를 위한 딥러닝'강좌를 한 번 들어보세요!
저도 그 강의를 통해 많이 늘었네요!
관련 영상은 유투브에 있습니다!
https://www.youtube.com/watch?v=BS6O0zOGX4E&list=PLlMkM4tgfjnLSOjrEJN31gZATbcj_MpUm
다음에 또 돌아오겠습니당~~!
'공대생 일기 > 프로젝트' 카테고리의 다른 글
Visual studio에서 OpenCV와 Real Sense SDK 설치해서 D415 연결해보기! (0) | 2019.07.08 |
---|---|
Jetson TX2에 Jetpack 설치하는 법 (0) | 2019.07.02 |