728x90 GPU1 tensorflow 2.X gpu 메모리 문제해결 tensorflow 2.X gpu 메모리 문제해결 gpus = tf.config.experimental.list_physical_devices('GPU')if gpus: try: # Currently, memory growth needs to be the same across GPUs for gpu in gpus: tf.config.experimental.set_memory_growth(gpu, True) tf.config.per_process_gpu_memory_fraction = 0.4 #메모리 사용율 제한 #logical_gpus = tf.config.experimental.list_logical_devices(.. 2020. 11. 16. 이전 1 다음 728x90