Caffe at Windows #02

Caffe at Windows #02

• 딥러닝프레임워크 Caffe 를 윈10에서 GPU 로 설치하는 법을 알아봅니다
  • 2018_05_22 에 확인된 Caffe 기준입니다
• 참고 링크
  CUDA Toolkit 8.0
  1. https://developer.nvidia.com/cuda-80-ga2-download-archive
  cuDNN Archive
  2. https://developer.nvidia.com/rdp/cudnn-archive

---

개요

• Caffe 를 빌드하는 데에 필수적인 프로그램에 대한 준비는 이전 글을 참고하세요
• Caffe 를 GPU 로 빌드하는 데에 필요한 프로그램은 다음 2가지입니다
  • CUDA Toolkit
    • 글쓴이는 CUDA 8.0 을 사용했습니다
  • cuDNN
    • 글쓴이는 cuDNN 5.0 을 사용했습니다

---

※ [ 주의 ] CUDA Toolkit 을 설치하기 이전에 반드시 Visual Studio 를 설치하기 바랍니다
※ CUDA Toolkit 설치과정에서 해당 시스템에 이미 설치되어 있는 Visual Studio 와의 연동을
※ 수행하기 때문에, CUDA Toolkit 부터 설치하면 연동 과정이 꼬여버려 추가적인 작업이 필요합니다

CUDA Toolkit 8.0

• [참고 링크] 1번에서 다운로드 받을 수 있습니다
  • 글쓴이는 Windows → x86_64 → 10 → exe (local) 로 진행했습니다

• 설치파일을 실행하면 위와 같은 화면을 봅니다
  • 잠시 기다리면 설치마법사가 등장합니다

• 동의 및 계속 을 클릭합니다

• 빠른 설치 를 선택합니다
• 다음 을 클릭합니다

• 설치가 진행됩니다
  • 잠시 기다리면 설치가 완료됩니다

• 설치가 완료되면 Windows + S 키를 입력합니다
• 검색창에서 환경 변수 를 입력합니다
• 검색 결과의 시스템 환경 변수 편집 을 클릭합니다

• 환경 변수 버튼을 클릭합니다
  • 환경 변수 창이 뜹니다

• 시스템 변수 그룹의 CUDA_PATH 변수를 확인합니다
  • 값이 .../CUDA/v8.0 인지 점검합니다
  • 아니라면 위와 같이 변경해줍니다

※ [ 주의 ] CUDA Toolkit 을 여러 버전으로 설치할 경우 CUDA_PATH 가 덮어씌여질 수 있습니다
※ Caffe 는 CUDA_PATH 라는 환경 변수로 CUDA Toolkit 에 접근하므로 이를 관리해야합니다

cuDNN 5.0

• [참고 링크] 2번에서 다운로드 받을 수 있습니다
  • 글쓴이는 cuDNN v5 Library for Windows 10 으로 진행했습니다

※ 다운로드를 하기 위해서는 nVidia Developer 회원가입이 필요합니다
※ 메일 인증만으로 쉽게 가입할 수 있으니 해당 과정은 생략합니다

• 다운로드 받은 zip 파일의 압축을 해제합니다

• 압축을 해제하면 cuda 라는 폴더를 얻습니다
  • 두번째 스크린샷은 해당 폴더의 내용입니다
• 이제 cuda 폴더를 임의의 위치에 둡니다

• 글쓴이는 C:/Library 로 옮기고 폴더명을 cuDNN 으로 변경했습니다

---

시작

• build_win.cmd 를 편집하는 단계 직전까지는 이전 글과 동일합니다

• Ninja 를 사용하지 않을 것이므로 해당 줄을 위와 같이 변경합니다

• Ninja 를 사용하지 않을 것이므로 해당 줄을 위와 같이 변경합니다

• cuDNN 을 사용할 것이기 때문에 해당 줄에 내용을 추가합니다
  • -DCUDNN_ROOT=[cuDNN location] ^ 을 입력합니다
  • 글쓴이의 경우 C:/Library 에 위치해 있고 폴더명이 cuDNN 이므로 위와 같이 작성했습니다

• cmd 를 실행하고 이전 글처럼 scripts\build_win.cmd 를 작성합니다
  • Enter 를 눌러 빌드를 시작합니다

• 빌드 설정이 성공적으로 반영되었는지 확인합니다
  • CUDA 항목에서 ver. 8.0 으로 인식됨을 볼 수 있습니다
  • cuDNN 항목에서 ver 5.0.5 로 인식됨을 볼 수 있습니다

• 빌드가 성공적으로 완료됨을 볼 수 있습니다
  • CPU 만으로 빌드할 때보다는 경과시간이 적게 나옵니다
  • 글쓴이의 경우 25 → 19 분

---

이상으로 Caffe 를 윈도우즈에서 빌드하는 과정을 알아보았습니다
감사합니다