편집
1,408
번
바뀜
잔글
→패키지 버전관련
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)
|}
|}