YOLO V2 환경 구축하기 1(CUDA 8.0과 CUDNN 5.1 설치)
안녕하세요? 오늘은 YOLO V2를 돌리기에 앞서 먼저 환경 구축을 해야하는데 환경 구축하는 방법에 대해 말해보겠습니다.
ubuntu 16.04(64bit)에 CUDA8.0과 cuDNN 5.1, OPENCV 3.3의 환경을 기반으로 YOLO V2를 설치해 사용했으며, CPU,GPU 사양은 다음과 같다.
CPU:Intel(R) Core(TM) i5-7600 CPU @ 3.50GHz
GPU:GeForce GTX 1070 8GB
먼저, ubuntu 16.04(64bit)에 CUDA8.0과 cuDNN 5.1의 설치에 대해 시작하겠습니다.
우분투를 처음 설치하고 나서 본격적인 설치 작업을 수행하기 전에 패키지 업데이트를 수행한다.
NVIDIA graphic driver를 설치하기 위해 다음과 같이 명령어를 입력한다.
완료되면 왼쪽 탭 메뉴에 톱니바퀴 모양의 아이콘으로 되어있는 System Settings에 들어가서 Software & Updates->
Additional Drivers 탭으로 이동한다.
CUDA 8.0은 최소 367.4x 이상의 드라이버가 필요하다. 나는 378로 설치했다.
그리고 스크롤을 내려보면 Using Processor microcode firmware for intel CPUs from intel-microcode(proprietary)라는 선택지가 있다. 같이 선택한 후 Apply Changes 버튼을 클릭해 설치한다. 설치가 끝나면 reboot를 해준다.
재부팅이 완료되고 터미널에서
를 입력하면 NVIDIA X Server Settings가 실행된다. 이러면 드라이버 설치가 완료된 것이다.
이제 CUDA Toolkit을 설치해보자. 설치 버전은 8.0이다.
회원가입을 하고 로그인을 한뒤,
Select Target Platform 메뉴에서 Linux->x86_64->Ubuntu->16.04->runfile(local)을 다운 받는다.
이 때, 다운받은 경로를 기억한 후 다운이 완료되면, runfile을 실행한다. 나같은 경우에서 home 디렉토리에
cuda라는 폴더를 다운로드 경로로 지정해놨다.
실행하면 End User License Agreement가 나오게 되는데 이때 컨트롤+c를 누르면, 다음과 같은 질문이 나온다.
설치가 진행되고 끝나면 Summary를 통해 결과를 확인할 수 있다. 우리가 확인해야할 summary는
이러한 내용이 나오면 설치가 정상적으로 완료된 것이다.
이제 PATH variable을 설정해야한다.
이렇게 설정해주면 되는데, 나같은 경우에는 YOLO를 실행시킬시에 경로 오류가 발생하여, 곤란을 겪었다. 다음 글에
오류를 보여줄테지만, 경로 자체를 vi ~/.bashrc에 들어가서, 입력시켰다. 그러면, 항상 경로가 유지되기 때문에 오류가
해결 되었다.
이렇게 출력되면 완료된 것이다.
https://developer.nvidia.com/rdp/cudnn-download에서 cuDNN v5.1과 CUDA 8.0에 해당하는 버튼을 누르고,
cuDNN v5.1 Library for Linux를 클릭해 다운받는다. 나와 같은 경우는 CUDNN이라는 빈 폴더를 만들어, 경로를 지정했다.
이 때 출력화면에 CUDNN_MAJOR가 포함되어 있으면 제대로 설치된 것이다.
이상으로, CUDA와 CUDNN 설치를 마치겠다.
다음 글에는 OPENCV 설치에 대해 쓰겠다.