바뀜

둘러보기로 가기 검색하러 가기
333 바이트 추가됨 ,  2022년 3월 21일 (월) 22:45
잔글
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)
 
|}
 
|}
  

둘러보기 메뉴