안녕하세요,
재재입니다.
이번에는 GPU 가속화 라이브러리인 cuda 를 쉽게 설치해볼게요.
(1) package 업데이트
먼저, key 등록을 진행할게요.
curl https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub
sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"
sudo apt-get update
(2) cuda 11.x 설치
아래는 11.0 을 설치하는 커맨드입니다.
설치하고자 하시는 버전에 맞게 0을 대체해주시면 됩니다.
# 11.0 installations
sudo apt-get -y install cuda-11-0
https://developer.nvidia.com/cuda-downloads
(3) cuda 환경변수 등록
# for 11.0
echo 'export PATH=/usr/local/cuda-11.0/bin${PATH:+:${PATH}}' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/lib:/usr/lib/x86_64-linux-gnu:/usr/local/cuda-11.0/lib64:{LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}' >> ~/.bashrc
같이 보면 좋을 컨텐츠 추천 드립니다.
cuda 11.x 설치 쉽게하기