Tensorflow-gpu 2.0 beta 윈도우 아나콘다 환경에 설치해봅시다!

2019. 6. 21. 10:57Programming Tutorial

튜토리얼식으로 쭉 따라가주시면 됩니다.

 

아래 몇가지 사실 먼저 알고 가시죠.

 

아나콘다 cuDNN version 7.6.0.을 지원함.

> CUDA 10.0이 필요함으로 GPU Drivers version 410.x 혹은 그이상이어야 합니다.

 

시작하기 전에 콘다에서 아직까진 공식적으로 지원하고 있지 않습니다.

아래방법을 말고 그냥 공식지원때까지 기다리는 것도 하나의 방법이겠네요.

 

그럼 시작합니다.

 

1. 아나콘다를 윈도우환경에 설치합니다.

https://www.anaconda.com/distribution/ 

 

Home - Anaconda

Latest news: Anaconda named a May 2019 Gartner Peer Insights Customers’ Choice for Data Science and Machine Learning Platforms. The Enterprise Data Science Platform for… Data Scientists Connect to a range of sources, collaborate with…

www.anaconda.com

 

2.설치가 완료되면 Anaconda Prompt를 열어봅시다.

 

현재 아나콘다는 cudnn 7.6.0을 tensorflow 1.13.1과 함께 지원하고 있습니다. 즉 공식지원이기 때문에 간단하게

가상환경 생성 하면서 설치까지 한번에 해볼수 있습니다. 아래와 같이 말이죠.

 

conda create -n your_env_name python=3.6 pip tensorflow-gpu

 

 

3. 한동안 쭈우욱 설치가 완료되면 생성된 가상환경에 들어가봅시다. 앞에 conda는 안붙여도 됩니다.

 

conda activate your_env_name

 

 

4. 그리고 tensorflow 2.0 베타를 쓸거니까. 2번단계에서 같이 설치된 tensorflow는 uninstall 해야죠.

 

conda uninstall tensorflow-gpu

 

 

 

5. 이제 마지막으로 공식지원하지 않는 tensorflow-gpu 2.0베타는 pip명령으로 설치합니다.

 

베타상태다 보니 설치방법이 바뀔수도 있어요. 

https://www.tensorflow.org/install/gpu#tensorflow_20_beta

 

GPU support  |  TensorFlow

TensorFlow GPU support requires an assortment of drivers and libraries. To simplify installation and avoid library conflicts, we recommend using a TensorFlow Docker image with GPU support (Linux only). This setup only requires the NVIDIA® GPU drivers. Thes

www.tensorflow.org

상기 웹페이지 꼭 확인해보고 설치 시작하죠.

 

pip install tensorflow-gpu==2.0.0-beta1

 

 

결론 : 자 이제 깔렸습니다. 이것저것 실험해보는게 남았네요.