새소식

반응형
IT skill/Python

cudnn failed to initialize 에러 해결방법

  • -
반응형

아래의 코드를 붙여 넣으면 된다.

 

from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession

config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)

 

끝!

반응형
Contents

포스팅 주소를 복사했습니다

이 글이 도움이 되었다면 공감과 광고 클릭 부탁드립니다~