Keras, Tensorflow에서 GPU 똑똑하게 사용하기 - 1부

Python/Keras 2018. 10. 11. 22:07

이번 포스팅에서는 Keras와 Tensorflow에서 GPU를 더 똑똑하게 사용하는 방법에 대해 알아보자. 케라스 (와 당연히 텐서플로우)를 사용한다면, GPU도 높은 확률로 사용 중일 것 이다. 근데 이놈의 텐서플로우는 default로 (2장 이상의 GPU를 사용한다면 모든) GPU의 메모리를 배정받으면서 시작되는데, 이 경우 파이썬 프로세스를 하나만 실행하기만 해도 GPU 메모리가 허덕이는 경우가 태반이다. 하는일은 하나도 없고 (util : 0%) 메모리는 95%를 먹고계신 Tensorflow ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[10000,32,28,28] [[Node: conv..

Article Thumbnail
1.1 윈도우에서 Tensorflow 설치하기 (GPU 버전)

Python/Tensorflow 2017. 3. 2. 17:07

2017/03/01 - [Tensorflow] - 1. 윈도우에서 Tensorflow 사용하기 (CPU 버전) 앞선 글에서는 윈도우에서 native pip을 사용하여 Tensorflow를 사용하는 방법을 알아보았다. 하지만 해당글에서 나온 설치방법만으로는 tensorflow-gpu를 제대로 사용할 수 없는데, 몇가지 툴셋들이 빠져있기 때문이다. 이번 글에서는 어떤것을 어떻게 깔아야 하는지 간략하게 알아보자. 1. 지금 상황은?필자의 Tensorflow 설치법을 따라서 tensorflow-gpu를 설치했다면, python에서 tensorflow를 불러올 때 아래와 같은 에러가 뜨게 된다. >>> import tensorflowI c:\tf_jenkins\home\workspace\release-win\de..

Article Thumbnail
1. 윈도우에서 Tensorflow 설치하기 (CPU 버전)

Python/Tensorflow 2017. 3. 1. 08:47

Tensorflow가 점차 업데이트됨으로써, 윈도우에서 docker나 다른 프로그램을 사용하지 않고도 tensorflow를 이용할 수 있게 되었다. 현재 공식홈페이지에는 native pip 설치와 Anaconda를 이용한 가상환경 구축으로 설치방법이 있지만, Anaconda는 2017년 3월 1일 기준 공식적으로 지원하는 설치방법이 아니므로, The Anaconda installation is community supported, not officially supported. 이번 포스팅에서는 윈도우에 native pip을 사용하여 tensorflow를 설치하는 방법에 대하여 알아보자 (※아마 이 글을 읽으며 터미널이 뭔지, 명령어는 뭘 쳐야하는지, 하나도 모르는 사람은 거의 없을 것 같지만, 그래도 이..

Article Thumbnail