주 메뉴 열기
대문
임의의 문서로
로그인
설정
Pywiki 소개
면책 조항
Pywiki
검색
바뀜
← 이전 편집
다음 편집 →
TensorFlow
(편집)
2022년 3월 21일 (월) 22:45 판
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)
|}
|}
Sam
사무관
,
인터페이스 관리자
,
관리자
, 교사
편집
1,408
번