Soohyun’s Machine-learning

텐서플로 GPU / 윈 10 설치 본문

Machine Learning/misc

텐서플로 GPU / 윈 10 설치

Alex_Rose 2017. 9. 17. 16:04

해당 가상 환경은 파이썬 3.5에서 했다. (activate tf)



1. 비주얼 스튜디오 2015를 설치한다. 


CUDA가 GTX 1050을 인식 못하는 문제가 있으나, 이 부분은 그냥 무시하고 지나가도 상관없는 듯 하다. 


몇 번을 깔아도 CUDA가 비주얼 스튜디오를 인식을 못하길래, 결국 해외 사이트를 구글링. 

CUDA 8.0 버전은 비주얼 스튜디오 2015를 지원한다고 해서 그걸로 깔았더니 

그제서야 드디어!! CUDA가 인식하고 넘어갔다.




2. cDNN 도 설치. 


3. 아나콘다를 설치할 때, PATH 등록 체크를 반드시 해주고 깐다. 

not recommended 이지만 GPU로 돌리려고 하면 필요한 작업인 듯 하다.


그리고 아나콘다 프롬프트로 conda install을 통해 가상 환경 작업이랑, 

numpy, pandas, scipy, scikit-learn, 텐서플로우 GPU버전 인스톨까지 전부 해줬다.

scipy는 계속 에러나서 whl 파일로 일일이 pip install 주소 쳐주고.whl 해줬어야 했는데 잘 깔리는 걸 보며 매우 신기해 함. 




4. Pycharm 설치



name: GeForce GTX 1050

major: 6 minor: 1 memoryClockRate (GHz) 1.493

pciBusID 0000:01:00.0

Total memory: 2.00GiB

Free memory: 1.60GiB 


ㅠㅠㅠㅠㅠㅠㅠㅠ 말을 잇지 못했다. 얼마나 기뻤던지... 

아직까지 텐서 CPU 버전과 성능 차이가 확 날 정도의 코드는 안 돌려봤다. 돌리면 많이 다르다는데.. 


설치하는데에 너무 고생해서 윈도우 컨테이너(도커)는 나중에 설치하는 걸로.... 


아예 노는 노트북 녀석을 리눅스 운영체제로 만들어버릴까도 생각중이다.

'Machine Learning > misc' 카테고리의 다른 글

용어 이해  (0) 2017.10.13
신경망 뼈대 구축하기  (0) 2017.10.13
머신러닝 잡다 정보  (0) 2017.10.09
머신러닝 자료 링크들  (0) 2017.10.09
신경망에 대해 이해한 것  (0) 2017.09.20
Comments