torch model.cuda() segmentation fault (core dumped) 에러 해결

Bert 관련 모델 학습 코드 실행 중 해당 오류가 발생했습니다.

(세그멘테이션 오류)

 

terminal 상에서는 모델 로드 및 model.cuda() 해도 에러가 안나서 찾아보다가 한참 걸렸네요.

 

해결 방법

코드 상 학습, 추론에 사용되는 모델은 torch였는데 "import tensorflow"가 선언되어있어 제거하니 정상 작동 되었습니다.

제 코드에서 tensorflow는 "from keras.preprocessing.sequence import pad_sequences" 이 소스만 사용하고 있어 굳이 "import tensorflow"를 할 필요가 없었습니다.