363번째 줄:
363번째 줄:
Function call stack: train_function -> train_function
Function call stack: train_function -> train_function
|텐서플로우, CUDA, CuDNN 등에서 뭔가 안맞았을 때...? 모르겠다 젠장;
|텐서플로우, CUDA, CuDNN 등에서 뭔가 안맞았을 때...? 모르겠다 젠장;
+
+
+
GPU자원의 문제라는 설도 있었다.
다 맞추고 텐서플로우를 GPU버전으로 다시 설치해도 뜨네;
다 맞추고 텐서플로우를 GPU버전으로 다시 설치해도 뜨네;
−
|
+
|GPU자원의 문제라면 탄력적으로 GPU를 사용하게 하는 방식으로 해결이 가능하리라...?
+
+
+
Tensorflow2.X의 경우.
+
+
import tensorflow as tf
+
+
config = tf.compat.v1.ConfigProto()
+
+
config.gpu_options.allow_growth = True
+
+
session = tf.compat.v1.Session(config=config)
|}
|}