본문 바로가기

computer setting/Deep learning

Deep learning setting - NVIDIA driver install (딥러닝 세팅 - 엔비디아 드라이버 세팅)

딥러닝 코드를 실행하기 위해서는 gpu가 필수! (gpu driver를 설치해줘야 사용할 수 있음)


세팅하고자 하는 컴퓨터 환경

  • Ubuntu 16.04
  • Intel i5-9400F 16GB RAM
  • GeForce GTX 1080 Ti 

NVIDIA driver install (엔비디아 드라이버 세팅)

1. software & update(소프트웨어 & 업데이트) 클릭

2. 추가 드라이버 클릭

3. 독점 드라이버 선택 > 다시 시작

* 드라이버 버전은 gpu 별로 여러 버전이 있으나,

   본인은 딥러닝 세팅하고자 하는 드라이버 버전이 384.xx 버전이었기 때문에, 해당 버전으로 설치

 

4.1 리부팅 후 동일한 경로로 들어가면 nvidia 드라이버가 제대로 설치된 걸 확인할 수 있음 or

4.2 터미널에서 하단의 명령어로 확인 가능

~$ nvidia-smi