파이참에 텐서플로우(TensorFlow GPU)설치하는 법

* TensorFlow 설치에 앞서 해당 버전에 필요한 python, cuDNN, CUDA 설치가 필요합니다.  

이 포스트는 여러 방법을 시행해보았지만 깔리지 않는 경우 시도하는 것을 추천합니다.

 

 

https://www.tensorflow.org/install/source_windows#tested_build_configurations

Windows의 소스에서 빌드  |  TensorFlow

 

GPU버전을 깔아주기 위해 해당 버전에 맞는 Python, cuDNN, CUDA를 설치해주시길 바랍니다.

 

 

 

Python 및 PyCharm 설치방법

 

- Python 설치방법: https://m.blog.naver.com/PostView.naverisHttpsRedirect=true&blogId=baekmg1988&logNo=221436032478 

[Python] 윈도우에 파이썬 다운로드 및 설치하는 방법

 

 

- PyCharm(파이참) 설치방법: https://dora-guide.com/pycharm-install/

파이참(PyCharm) 다운로드 및 설치 후 설정 방법 - 도라가이드

 

- cnDNN, CUDA 설치 방법: https://n1094.tistory.com/42

[tensorflow-gpu 설치] 2. CUDA + cuDNN 설치하기

---

Python과 PyCharm이 설치되어있다는 가정하에 진행하겠습니다.

 

 

1. 파이참 실행 후 Settings 를 들어간다.

2. Python interpreter 창에 들어간다. 그리고 +버튼 클릭

 

 

3. 클릭하면 이 창이 뜬다.

 

4. tensorflow-gpu 를 검색한 뒤 install 버튼 클릭

 

5. 설치완료 후 테스트

import tensorflow as tf

print(tf.__version__)

새 파이썬 파일을 만들어 위 코드를 작성한 뒤 실행시킨다.

 

버전이 뜨면 잘 설치가 된 것이다.