colab이아닌 개인 컴퓨터의 하드웨어로 훈련시 오류

[Yunjong Lee]

colab이 아니라 주피터노트북에서 모델을 만들고 훈련할 시 오류가 뜨는데 혹시 

이와 관련하여 원인이 무엇인지 여쭙고싶어 문의드립니다.

현재 그래픽카드는 rts 2060super를 사용중이고 

cpu로 훈련을 진행할시엔 문제가 없이 잘 훈련이 진행되는데

gpu로 훈련할시에만 오류가 뜹니다.

오류의 내용은

Node: 'gradient_tape/sparse_categorical_crossentropy/SparseSoftmaxCrossEntropyWithLogits/UnsortedSegmentSum' 

Deterministic GPU implementation of unsorted segment reduction op not available.     [[{{node gradient_tape/sparse_categorical_crossentropy/SparseSoftmaxCrossEntropyWithLogits/UnsortedSegmentSum}}]] [Op:__inference_train_function_31734]

입니다.

혹시 책의 딥러닝 부분의 모델은 개인 GPU가 아닌 코랩에서만 훈련이 가능한 자료인가요? 

[Haesun Park]

안녕하세요. 박해선입니다.

어느 책의 코드인지 모르겠지만 코랩에서 문제없이 실행되면 다른 GPU에서도 실행 가능합니다.

만약 다른 GPU에서 실행이 안된다면 고수준 텐서플로 코드(케라스)가 아니라 

저수준 커널 구현이나 CUDA, cudnn 라이브러리에 관련된 문제입니다.

이런 문제는 텐서플로 깃허브나 엔비디아 포럼에 문의하시면 좋습니다.

다음은 제가 찾은 비슷한 오류에 관한 이슈입니다.

https://github.com/tensorflow/tensorflow/issues/54276

감사합니다!

2022년 8월 23일 (화) 오후 8:15, Yunjong Lee <leeyj...@gmail.com>님이 작성:

--
이 메일은 Google 그룹스 '머신러닝/딥러닝 도서 Q&A' 그룹에 가입한 분들에게 전송되는 메시지입니다.
이 그룹에서 탈퇴하고 더 이상 이메일을 받지 않으려면 ml-dl-book-qn...@googlegroups.com에 이메일을 보내세요.
웹에서 이 토론을 보려면 https://groups.google.com/d/msgid/ml-dl-book-qna/ac5ffd50-8283-429f-9df1-cdb2ea9dbb40n%40googlegroups.com을(를) 방문하세요.