편집
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)
|}
|}