티스토리 뷰

https://github.com/matterport/Mask_RCNN/issues/2510

 

failed to run cuBLAS routine: CUBLAS_STATUS_EXECUTION_FAILED while training during Epoch 1/1 · Issue #2510 · matterport/Mask_R

Error Logs Epoch 1/1 C:\Users\DanielSun\PycharmProjects\FRC-opencv-python\venv\lib\site-packages\skimage\transform\_warps.py:830: FutureWarning: Input image dtype is bool. Interpolation is not defi...

github.com

 

해당 답변을 참고하니, RTX 30XX버전의 GPU사용을 위한 CUDA와 tensor flow 1.X 버전의 호환성 문제라고 한다. 해결 방법은 Docker를 사용하거나, 2000번대 이하의 CUDA 10.X버전을 사용하는 그래픽카드에서 환경을 구축하는 방법밖에는 없어보인다.

Docker는 도저히 공부할 용기가 안나서 그냥 2080Ti를 사용하는 서버를 얻어서 쓰기로 했다.

 

추가로 케라스 공식 블로그를 참고하니 tensorflow 1.X 버전과 호환되는 keras버전을 잘 확인하고 설치해야 한다.

keras 2.3.X 버전부터는 tensorflow 2.0이상에서만 지원되는 것 같다.

댓글