주요 콘텐츠로 건너뛰기
버전: 1.0.0

Linux에 CUDA 설치

퍼블릭 클라우드에서 Nvidia 지원 VM 인스턴스를 사용하는 경우, 클라우드에서 제공하는 VM 이미지를 사용해야 합니다. 일반적으로 올바른 버전의 Nvidia 드라이버와 CUDA 툴킷이 이미 설치되어 있습니다. 자체 머신에 Nvidia 드라이버 및 CUDA 툴킷을 설치해야 하는 경우 이 글을 읽어보세요.

우분투 22.04​

1 Nvidia 드라이버를 설치합니다.​

그럽 구성을 다시 빌드합니다:

sudo apt-get install -y gcc make linux-headers-$(uname -r)
cat << EOF | sudo tee --append /etc/modprobe.d/blacklist.conf
blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist nvidiafb
blacklist rivatv
EOF
sudo sed -i 's/GRUB_CMDLINE_LINUX=""/GRUB_CMDLINE_LINUX="rdblacklist=nouveau"/' /etc/default/grub
sudo update-grub

Nvidia 드라이버 다운로드 및 설치

wget https://storage.googleapis.com/nvidia-drivers-us-public/GRID/vGPU16.1/NVIDIA-Linux-x86_64-535.104.05-grid.run
sudo sh NVIDIA-Linux-x86_64-535.104.05-grid.run

드라이버가 성공적으로 설치되었는지 확인

엔비디아-에스미 -q | 헤드

==============NVSMI LOG==============

타임스탬프 : 금 Oct 27 21:54:05 2023
드라이버 버전 : 535.104.05
CUDA 버전 : 12.2

연결된 GPU : 1
GPU 00000000:00:1E.0
    제품 이름 : NVIDIA A10G

GSP를 비활성화하고 재부팅합니다.

sudo touch /etc/modprobe.d/nvidia.conf
echo "옵션 nvidia NVreg_EnableGpuFirmware=0" | sudo tee --append /etc/modprobe.d/nvidia.conf
sudo 재부팅

2 CUDA 툴킷을 설치합니다.​

wget https://developer.download.nvidia.com/compute/cuda/12.2.2/local_installers/cuda_12.2.2_535.104.05_linux.run
sudo sh cuda_12.2.2_535.104.05_linux.run --silent --override --toolkit --samples --toolkitpath=/usr/local/cuda-12 --samplespath=/usr/local/cuda --no-opengl-libs

CUDA가 설치되어 있는지 확인합니다.

/usr/local/cuda/bin/nvcc --version

nvcc: NVIDIA (R) 쿠다 컴파일러 드라이버
저작권 (c) 2005-2023 NVIDIA Corporation
빌드_Aug_15_22:02:13_PDT_2023에 구축됨
쿠다 컴파일 도구, 릴리스 12.2, V12.2.140
Build cuda_12.2.r12.2/compiler.33191640_0

그런 다음 다음 두 명령을 사용하여 환경 경로를 설정합니다. 아마도 다음 두 줄을 ~/.bashrc 그리고 ~/.bash_profile (또는 ~/.zshrc 그리고 ~/.profile) 파일을 삭제하여 새 터미널과 향후 로그인 시에도 이러한 CUDA 라이브러리 파일을 찾을 수 있도록 합니다.

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

추가 리소스​

문제가 발생할 경우 도움이 될 수 있는 스크립트가 더 있습니다.

행운을 빕니다!