openxlab

mmdetection3d 실행 준비 & 쿠다 설치 & nvidia driver 설치 -1

wandering developer 2024. 7. 15. 21:05

1. Nuscene mini 다운로드 하기

mkdir -p /data/sets/nuscenes
wget https://www.nuscenes.org/data/v1.0-mini.tgz

tar -xf v1.0-mini.tgz -C /data/sets/nuscenes

 

2.1 cuda  버전 확인 절차 프로세스

https://mmdetection3d.readthedocs.io/en/latest/get_started.html

 

기본적으로 그래픽카드에 따라 버전이 달라짐.

 

그래픽 카드 확인

" nvidia-smi --query | fgrep 'Product Name'"

 

나의 경우는 " NVIDIA GeForce GTX 1650 with Max-Q Design"

 

아래 주소에서 코어 확인 가능

https://en.wikipedia.org/wiki/CUDA

 

CUDA - Wikipedia

From Wikipedia, the free encyclopedia Parallel computing platform and programming model In computing, CUDA (originally Compute Unified Device Architecture) is a proprietary[1] parallel computing platform and application programming interface (API) that all

en.wikipedia.org

 

위 주소에서 확인해보면 내 노트북은 Turing 아키텍쳐 인것을 확인 할 수 있고

 

튜링 아키텍쳐는 10.0 버전 부터 설치가 가능한 것을 확인 할 수 있다.

 

나의 경우 https://github.com/youngskkim/CRN/tree/main 저자가 11.1을 사용하고 위에서 난 11.1 을 설치할 수 있으므로 11.1을 설치할 것이다.

 

2.2 cuda  버전 삭제

우선 기존에 있는 cuda와 nvidia를 지운다.

#!/bin/bash

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*'
sudo dpkg -l | grep nvidia
echo 'sudo dpkg -l | grep nvidia'
sudo dpkg -l | grep cuda
echo 'sudo dpkg -l | grep cuda'

 

2.3 쿠다 11.1 설치

우분투 버전을 고려해서 아래 주소를 이용해서 설치한다.

https://developer.nvidia.com/cuda-11.1.1-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=deblocal

 

CUDA Toolkit 11.1 Update 1 Downloads

Get CUDA Toolkit 11.1 Update 1 for Linux and Windows.

developer.nvidia.com

 

여기서 주의 해야할 점은

sudo apt-get -y install cuda 대신에

sudo apt-get -y install cuda-toolkit-11-1

 

을 설치한다. cuda를 설치할 경우 nvidia 그래픽 카드가 너무 옟날 꺼여서 설치가 안되는 경우가 있다.

 

그래서 따로 설치한다.

 

2.4 nvidia  드라이버 설치 

예날 버전 중 안정적인 버전 하나 선택해서 설치함.

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update

# 적절한 NVIDIA 드라이버 버전 확인
ubuntu-drivers devices

# 적절한 버전 설치
sudo apt-get install nvidia-driver-470

 

~/.bashrc 에 아래 내용 추가함.

export LD_LIBRARY_PATH_=/usr/local/cuda/lib64:{LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export PATH=/usr/local/cuda/bin:${PATH:+:${PATH}}

 

재부팅 하면

 

2.5 확인 아래쳐럼 나오면 잘 된것이다.

$nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Mon_Oct_12_20:09:46_PDT_2020
Cuda compilation tools, release 11.1, V11.1.105
Build cuda_11.1.TC455_06.29190527_0

, nvidia-smi 쳤을때 

$nvidia-smi
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 470.256.02   Driver Version: 470.256.02   CUDA Version: 11.4     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  NVIDIA GeForce ...  On   | 00000000:02:00.0 Off |                  N/A |
| N/A   40C    P8     2W /  N/A |    673MiB /  3914MiB |      5%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+

 

이제 여기까지 되었으면 https://github.com/youngskkim/CRN 프로젝트를 실행하기 위한 준비가 다 된것이다.

반응형