위성 이미지 객체 검출 미래도전기술 경진대회

안녕하세요. Object detection API를 실행하면서 여러가지 오류가 발생할 수 있는데,

그 중 하나인 GPU-CUDA-TENSORFLOW 버전 호환성에 대해서 다루어 보려고 합니다.

1. GPU - CUDA 호환성
2. NVIDA GPU는 micro-architecture에 따라 세대가 구분되는데, 유명한 과학자들의 이름을 따서 Tesla, Fermi, Kepler, Maxwell, Pascal, Volta, Turing 으로 구분이 됩니다.
3. GPU는 각 각의 세대에 맞는 Compute capability를 가지고 있는데 이 Compute capability 별 지원되는 CUDA 버전이 다르게 구성되어 있습니다
4. 다음의 페이지로 이동하시어 가지고 있는 GPU 모델에 맞는 Compute capability를 확인해주시고 이에 맞는 CUDA 버전을 설치해주시기를 바랍니다.  

                       GPUs supproted

                  d. CUDA 버전에 맞는 최소 GPU 드라이버 버전이 있으며, 이에 맞게 드라이버를 업데이트 해주셔야 합니다.

ex) RTX 2060 - Compute capability: 7.5, micro-architecture: Turing

• CUDA 10.0 - 10.2 호환 가능, 9.0 - 9.2 호환 제한

   ※ CUDA 10.x 버전은 대부분의 Compute capability를 수용하며 CUDA 9.0은 일부 최신 GPU 모델을 지원하지 않습니다.

       2.CUDA - TENSORFLOW 호환성

a. GPU에 적합한 CUDA를 설치하셨으면, 이에 맞는 Tensorflow 버전을 선택하여 설치를 해주셔야 합니다.

            b. 해당 내용은 https://www.tensorflow.org/install/gpu 에서 확인할 수 있습니다.

       3.구글 코랩 기준(CUDA 10.0, tensorflow 1.15.x-gpu 버전) 정상적으로 baseline 코드가 실행됨을 확인하였으며,

CUDA 9.0, tesnorflow 1.6.0-gpu 에서도 정상적으로 돌아감을 확인하였습니다.

해당 사항 참고하시어 오류 해결에 도움이 되시길 바라겠습니다.

감사합니다.