2021년 4월 10일 토요일

가장 쉬운 Ubuntu 18.04 NVIDIA driver, CUDA and Tensorflow installation

이 글은 Ubuntu 18.04 NVIDIA driver, CUDA and Tensorflow 설치 방법에 대한 것을 간략히 정리한 것이다. 그동안 18.04 버전 설치 테크트리가 약간 변경되어 다시 재정리한다. 다만, 딥러닝 등 최신 라이브러리 사용 목적이라면 우분투 20.04를 권장한다.
  • Ubuntu 18.04 설치 이미지 부팅 가능 USB 만들기
  • Install Ubuntu 18.04. 만약, 화면이 안보이거나 멈추면 GRUB에서 NOMODESET 추가 후 설치 진행
  • Booting Ubuntu
  • Ubuntu에서 Terminal을 열어 아래 명령어 입력 실행함 
    • ubuntu-drivers devices
    • sudo ubuntu-drivers autoinstall
    • reboot
    • nvidia-smi
  • Install CUDA 10.2 and 11.2 or 11.0. 주의: 이 단계에서 GPU 드라이버 재 설치하면 안됨.
  • Install cuDNN SDK 8.0.4
  • Install Tensorflow
  • Install Keras
  • Install Jupyter Notebook
  • reboot
이제, 다음과 같이 딥러닝 예제들이 정상 실행되면 성공한 것이다.
결과

참고로, Ubuntu 18.04 현재 버전은 ROS 설치를 완벽히 지원하지 않는다. 디펜던시 에러가 발생하므로 주의한다.

댓글 없음:

댓글 쓰기