[환경설정] CUDA 드라이버 설치 (in Ubuntu 16.04)

Ubuntu 16.04에 CUDA 드라이버 설치

 

지난번 포스팅 2021.11.24 - [Setting] - [환경설정] NVIDIA 드라이버 설치 (in Ubuntu 16.04) 에 이어서 이번에는 CUDA 드라이버까지 설치를 진행할 예정이다. CUDA를 설치하기 전에 NVIDIA 드라이버 버전을 확인하고 CUDA와 호환이 되는지 확인 후 설치하는게 가장 좋다.

 

1. CUDA Driver 다운로드

• NVIDIA 드라이버 다운로드와 마찬가지로 홈페이지에서 버전을 선택하여 설치한다.
• 아래 표를 보고 버전 호환 여부를 체크하고 설치하자!

• 

   
• 설치를 하다보면 Installer Type으로 runfile(local), deb(local), deb(network), cluster(local) 네가지의 옵션이 있다. 그중 반드시 runfile(local) 로 설치를 한다. deb로 설치 시 GPU 드라이버의 설치 유무를 확인하지도 않고 재설치를 수행함으로 기존에 설치한 GPU 드라이버와 충돌을 이르킨다.

 

2. CUDA Toolkit 설치

• NVIDIA 드라이버 설치와 마찬가지로 우선 GUI 서비스를 종료 시키고

  • $ sudo service lightdm stop

• 1번에서 다운로드한 CUDA Toolkit 파일을 실행한다. version은 본인이 설치한 버전으로

  • $ sudo sh cuda_version_linux

• NVIDIA 드라이버 때는 계속 넘어갔지만 CUDA 에서는 반드시 확인해야한다.

  •  Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 387.26?
    (y)es/(n)o/(q)uit:​

     
  • 처럼 문구가 나오면 반드시 No
    • 가장 많이 실수하는 부분으로, 반드시 No를 선택한다. 혹, nvidia-xconfig를 설치하길 원하는지 물을 때가 있다. GPU 드라이버 설치 시 이미 설치하였으므로 이 경우도 반드시 No를 선택한다. (그렇지 않으면 기존 설치된 드라이버와 충돌이 일어날 수도 있다.)
  • 이후 계속 Yes / Ok
• 설치가 완료되었으면 다시 GUI 서비스를 재실행 하고

  • $ sudo service lightdm start

• 시스템을 재부팅 하면 설치 완료

  • $ sudo reboot -n

3. profile 수정

• 아래 내용을 .profile에 추가한다.

  • if [ -d "/usr/local/cuda/bin" ]; then
    PATH="/usr/local/cuda/bin:$PATH"
    fi
    if [ -d "/usr/local/cuda/lib64" ]; then
    LD_LIBRARY_PATH="/usr/local/cuda/lib64":$LD_LIBRARY_PATH
    fi
    
    export PATH 
    export LD_LIBRARY_PATH

• 그다음 설치 확인을 하면 설치가 된것을 확인할 수 있다.

  • nvcc -V

  •