Ubuntu 18.04에 NVIDIA driver+CUDA Toolkit+cuDNN 설치
4383 단어 CUDANVIDIAubuntu18.04우분투
소개
심층 학습 프로그램을 GPU에서 실행하는 데 필요한 NVIDIA Driver, CUDA Toolkit 및 cuDNN을 설치하십시오.
환경
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
아마도 여기 두 가지로 나뉩니다.
sudo ./NVIDIA-Linux-x86_64-430.50.run --no-x-check
를 실행하여 다시 시작합니다.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
완료.
참고 사이트
Reference
이 문제에 관하여(Ubuntu 18.04에 NVIDIA driver+CUDA Toolkit+cuDNN 설치), 우리는 이곳에서 더 많은 자료를 발견하고 링크를 클릭하여 보았다 https://qiita.com/fury00812/items/f836583f5f2030d24fcd텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
우수한 개발자 콘텐츠 발견에 전념 (Collection and Share based on the CC Protocol.)