Ubuntu 18.04에 NVIDIA driver+CUDA Toolkit+cuDNN 설치

소개



심층 학습 프로그램을 GPU에서 실행하는 데 필요한 NVIDIA Driver, CUDA Toolkit 및 cuDNN을 설치하십시오.

환경


  • OS: Ubuntu 18.04
  • GPU: Nvidia Geforce GTX960

  • 1. Nvidia 드라이버 설치



    공식 사이트 에서 드라이버를 검색, 다운로드. 자신의 경우 NVIDIA-Linux-x86_64-430.50.run
    실행 권한을 부여하여 설치
    cd Downloads/
    chmod +x NVIDIA-Linux-x86_64-430.50.run
    sudo ./NVIDIA-Linux-x86_64-430.50.run --no-x-check
    

    여기서, 기본 드라이버 nouveau와 충돌하면 화난


    말하는대로 진행하면 실패합니다.


    lsmod로 확인하면 ...
    lsmod | grep nouveau
    



    아직 nouveau가 비활성화되지 않았 음을 알 수 있습니다.
    업데이트하고 다시 시작하십시오.
    sudo update-initramfs -u
    reboot
    

    아마도 여기 두 가지로 나뉩니다.
  • 잘 재부팅 할 수 있고, nouveau가 비활성화되었음을 확인할 수있는 사람은 다시 sudo ./NVIDIA-Linux-x86_64-430.50.run --no-x-check를 실행하여 다시 시작합니다.
  • 드라이버를 비활성화 한 후 화면 어두운 사람은 복구 모드를 시작하고 (참고 URL) 루트로 shell에 들어가서 sudo ./NVIDIA-Linux-x86_64-430.50.run --no-x-check를 실행하여 다시 시작합니다.
    nvidia-smi
    

    통과하는지 확인하고 완료하십시오.

    2. CUDA Toolkit 설치



    여기 에서 다운로드. 버전 10.0 dev(local)을 선택했습니다.

    설치.
    cd Downloads/
    sudo dpkg -i cuda-repo-ubuntu1804-10-0-local-10.0.130-410.48_1.0-1_amd64.deb
    sudo apt-key add /var/cuda-repo-10-0-local-10.0.130-410.48/7fa2af80.pub
    sudo apt-get update
    sudo apt-get install cuda-toolkit-10-0
    

    패스를 통과하십시오.
    echo -e "# cuda paths" >> ~/.bash_profile
    echo 'export PATH="/usr/local/cuda-10.0/bin${PATH:+:${PATH}}"' >> ~/.bash_profile
    echo 'export LD_LIBRARY_PATH="/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}"' >> ~/.bash_profile
    source ~/.bashrc
    

    다음 명령으로 cuda 버전을 확인할 수 있으면 완료됩니다.
    nvcc -V
    

    3. cuDNN 설치



    여기 에서 회원등록 후, 다운로드.
    "cuDNN Runtime Library for Ubuntu18.04 (Deb)", "cuDNN Developer Library for Ubuntu18.04 (Deb)", "cuDNN Code Samples and User Guide for Ubuntu18.04 (Deb)"
    sudo dpkg -i libcudnn7_7.4.2.24-1+cuda10.0_amd64.deb
    sudo dpkg -i libcudnn7-dev_7.4.2.24-1+cuda10.0_amd64.deb
    sudo dpkg -i libcudnn7-doc_7.4.2.24-1+cuda10.0_amd64.deb
    

    완료.

    참고 사이트


  • How To Install Nvidia Drivers and CUDA-10.0 for RTX 2080 Ti GPU on Ubuntu-16.04/18.04
  • CUDA Toolkit 8.0을 Ubuntu Server16.04LTS에 설치하려고하면 Nouveau 커널 드라이버가 방해 받고 실패합니다.
  • 자작 PC에 Ubuntu16.04 LTS, NVIDIA 드라이버, CUDA9.0, cuDNN 7.1을 설치(딥 러닝용)
  • Ubuntu 16.04를 설치하고 NVIDIA 드라이버 (410.48)와 CUDA10.0 및 cuDNN7.4.1을 넣고 Tensorflow-gpu를 이동하는 메모
  • 좋은 웹페이지 즐겨찾기