<aside> 💡 대호선배 추천 설치 순서

  1. Graphic driver > Download driver from NVIDIA official site
  2. Anaconda (Open source distribution for python)
  3. CUDA (GPU Accelerator) : from NVIDIA official site
  4. CuDNN (GPU accelerated library)
  5. Deep learning frameworks
  6. PyCharm: Editor </aside>

Nvidia-driver-470 설치

[Ubuntu 18.04] 딥러닝 환경 구축 CUDA, cuDNN 설치

$ ubuntu-drivers devices

→nvidia-driver-470 추천하고있음 →이걸로 설치


CUDA 설치

CUDA Toolkit 11.4 Downloads

nvidia-smi에 나오는 cuda version에 맞춰 11.4 설치

$ wget <https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.42.01_linux.run>
$ sudo sh cuda_11.4.0_470.42.01_linux.run
$ sudo vi ~/.zshrc

# 파일 맨 밑에 추가
export CUDA_HOME=/usr/local/cuda-11.1
export PATH=/usr/local/cuda-11.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64:$LD_LIBRARY_PATH

$ source ~/.zshrc

$ nvcc -V 로 설치 확인