Ubuntu 18.04에서 CUDA, Cudnn 설치

데스크탑에서 Yolo를 사용하려면 CUDA와 Cudnn을 설치해줘야 합니다.

 

Jetson은 Jetpack을 설치할 때 알아서 다 깔리지만 데스크탑은 따로 설치를 해줘야 해서 좀 귀찮..

---

먼저 데스크탑에 이미 설치된 옛날 버전의 cuda를 지워줍니다.

sudo apt-get --purge remove 'cuda*'
sudo apt-get autoremove --purge 'cuda*'

이렇게도 지우고

cd /usr/local

다음 경로로 이동해서 cuda이름이 들어간 모든 파일을 지웁니다.

 

sudo rm -rf "파일명" 명령어로 삭제해도 되고 직접 폴더 경로로 들어가 지워도 됩니다.

 

home에도 cude 폴더가 있다면 모두 지워줍니다.

---

이제 CUDA를 설치해줍니다.

https://developer.nvidia.com/cuda-toolkit-archive

CUDA Toolkit Archive

원하는 버전을 선택하고 자신의 컴퓨터 세팅에 맞게 눌러주면

다음과 같이 설치할 때 필요한 명령어가 나옵니다.

 

저는 제일 최신버전인 11.7으로 설치했습니다.

 

그리고

gedit ~/.bashrc

로 bashrc 파일을 열어 맨 밑에

export CUDA_HOME=/usr/local/cuda-11.7
export PATH=/usr/local/cuda-11.7/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.7/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

다음과 같이 경로를 추가해줍니다. 11.7버전이 아닌 다른 버전을 설치했다면 거기에 맞게 수정해줍니다.

 

nvcc --version

다음 명령어로 정상적으로 설치되었는지 버전은 몇인지 확인이 가능합니다.

---

다음으로 cudnn을 설치합니다.

 

https://developer.nvidia.com/rdp/cudnn-download

 

여기서 자신이 설치한 CUDA 버전에 맞는 cudnn을 선택합니다. (NVIDIA 로그인 필수)

 

저는 11.7이므로 11.x에 맞는 8.5를 선택했고 그 중에서 

Linux x86_64 Tar 파일을 다운 받아줍니다.

 

그 다음 Download 폴더에 들어가 다운 받은 Tar 파일의 압축을 풀어줍니다.

 

그리고 다음 명령어로 압축을 푼 파일에 내용을 위에서 설치한 CUDA 파일에 넣어줍니다.

cd Downloads/cudnn-linux-x86_64-8.5.0.96_cuda11-archive/
sudo cp include/cudnn* /usr/local/cuda-11.7/include
sudo cp lib/libcudnn* /usr/local/cuda-11.7/lib64/
chmod a+r /usr/local/cuda-11.7/lib64/libcudnn*

 

정상적으로 설치가 되었는지 확인해줍니다.

cat /usr/local/cuda-11.7/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

 

다음과 같은 내용이 출력된다면 설치 완료입니다!

#define CUDNN_MAJOR 8
#define CUDNN_MINOR 5
#define CUDNN_PATCHLEVEL 0

---

cuda11.3에 맞는 파이토치 버전 설치

pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu11

 

이제 yolo를 설치할 수 있게 되었습니다 :)