728x90
반응형
■ 텐서플로 GPU 버전 실행시 세션 설정하기
------------------------------------------------------------------------------------------------------------------------
import tensorflow as tf
...
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
with tf.Session(config = config) as session:
...
------------------------------------------------------------------------------------------------------------------------
728x90
반응형
'Python > tensorflow' 카테고리의 다른 글
[PYTHON/TENSORFLOW] 텐서플로 GPU 버전 실행시 세션 설정하기 (0) | 2019.05.25 |
---|---|
[PYTHON/TENSORFLOW] 파이썬 텐서플로 GPU 버전 설치하기 (0) | 2019.05.25 |
[PYTHON/TENSORFLOW] 파이썬 텐서플로 GPU 버전 설치하기 (0) | 2018.09.26 |
[PYTHON/TENSORFLOW] 양방향 순환 신경망 만들기 (MNIST) (0) | 2018.08.13 |
[PYTHON/TENSORFLOW] 순환 신경망 만들기 (MNIST) (0) | 2018.08.13 |
[PYTHON/TENSORFLOW] tensorflow 모듈 : __version__ 변수를 사용해 버전 구하기 (0) | 2018.08.12 |
[PYTHON/TENSORFLOW] 컨볼루션 오토 인코더 만들기 (0) | 2018.08.10 |
[PYTHON/TENSORFLOW] 노이즈 제거 오토 인코더 만들기 (0) | 2018.08.07 |
[PYTHON/TENSORFLOW] 오토 인코더 만들기 (0) | 2018.08.07 |
[PYTHON/TENSORFLOW] 컨볼루션 신경망 만들기 (얼굴 감정 판정) (0) | 2018.08.05 |
[PYTHON/TENSORFLOW] 컨볼루션 신경망 만들기 (MNIST) (0) | 2018.08.04 |
댓글을 달아 주세요