[Ubuntu] Detectron2 & CUDA 설치(엔비디아 그래픽 드라이버 버전 선택 설치)

조경호·2022년 6월 16일
0

Ubuntu

목록 보기
2/2

우분투 20.04 버전에서 Detectron2를 설치하고 싶은데 설치 가능한 nvidia 그래픽 카드 드라이버 버전이 Detectron2 에서 요구하는 CUDA 버전이랑 맞는게 없다..

Detectron2에서 CUDA 11.3 또는 11.1 버전을 요구하는데..
CUDA 버전에 맞는 엔비디아 드라이버가 450(CUDA 11.1) 또는 465(CUDA 11.3) 버전이 있다..

우분투 16 버전에서는 450이든 460이든 465버전이든 아래 명령어로 버전을 선택하여 설치가 가능했었다..

sudo apt install nividia-driver-465

그런데 언젠가부터 설치 가능한 드라이브 목록에 뜨질 않더니 위 명령어로 버전 선택해서 설치해도 드라이버가 470 이상 버전으로 자동 업그레이드 되어 설치가 되었다.. 이게 무슨..

그래픽 카드가 최신(RTX 3080ti & RTX 3070ti)이라 그런건.. 아니었다..
GTX 1060도 안된다 짜증나게..

드라이버가 Deprecated 된건가?

일단 클린한 상태로 만들자

sudo apt-get purge nvidia*
sudo apt-get autoremove
sudo apt-get autoclean

sudo rm -rf /usr/local/cuda*
sudo apt-get --purge remove 'cuda*'
sudo apt-get autoremove --purge 'cuda*'

구글에 엔비디아 고급 드라이버 검색을 검색하자

고급 드라이버 검색에 들어가서 그래픽카드에 맞는 드라이버를 찾자

원래는 전체로 검색하면 465 버전이 떴었는데.. 이제는 또 안뜬다..

검색 필터 중에 최신: <= 이부분을 New Feature Branch로 선택하자.. 그럼 465 버전이 드디어 보인다..

다운로드한 465 드라이버를 설치하자

설치 방법은 검색하면 나온다..

쿠다 툴킷 아카이브로 가자

설치 방법이 3개가 있는데 로컬 설치 버전으로 받자..

  • 설치 후 PATH 설정 관련 인포가 뜨면 설치가 성공한 것이다..

쿠다 버전에 맞게 PATH 설정을 해주자..
~/.profile 알제?

export PATH=$PATH:/usr/local/cuda-11.3/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.3/lib64
export CUDADIR=/usr/local/cuda-11.3

디텍트론 설치는 공식 문서 따라하면 간단하다

conda 환경 생성 후 torch 부터 conda install 을 통해 버전에 맞게 설치하자..
해당 환경에서 torch 버전 찍어보고 문제가 없으면 디텍트론2를 설치하자..

파이썬 라이브러리 에러

뭔가 업데이트가 된 것 같다.. 아래 에러가 발생한다..

attributeerror: module 'distutils' has no attribute 'version'

라이브러리를 하나 설치해주자..

pip install setuptools==59.5.0
profile
개발자

0개의 댓글