tensorflow에서 GPU안쓰기

CUDA버전 tensorflow를 실행하면 GPU가 자동으로 활성화된다

학습할 때에는 GPU를 이용하면 훨씬 빠르지만, GPU 단점이 초기 설정하는데 시간이 오래걸린다

테스트할때에는 계산량이 비교적 적어서 GPU를 끄는 편이 더 빠르다

config = tf.ConfigProto(
        device_count = {'GPU': 0}
    )
sess = tf.Session(config=config)

위 코드를 넣어주면 GPU를 비활성화 한다

Advertisements

답글 남기기

아래 항목을 채우거나 오른쪽 아이콘 중 하나를 클릭하여 로그 인 하세요:

WordPress.com 로고

WordPress.com의 계정을 사용하여 댓글을 남깁니다. 로그아웃 / 변경 )

Twitter 사진

Twitter의 계정을 사용하여 댓글을 남깁니다. 로그아웃 / 변경 )

Facebook 사진

Facebook의 계정을 사용하여 댓글을 남깁니다. 로그아웃 / 변경 )

Google+ photo

Google+의 계정을 사용하여 댓글을 남깁니다. 로그아웃 / 변경 )

%s에 연결하는 중