본문 바로가기

자료

YOLO(Darknet) - 설치(GPU 설정)

728x90

1. 본인 GPU 확인 


2. CUDA 지원, 성능, 코어 수 확인

https://developer.nvidia.com/cuda-gpus

 

CUDA GPUs

Recommended GPU for Developers NVIDIA TITAN RTX NVIDIA TITAN RTX is built for data science, AI research, content creation and general GPU development. Built on the Turing architecture, it features 4608, 576 full-speed mixed precision Tensor Cores for accel

developer.nvidia.com

컴퓨터 계산능력 값 확인


3. CUDA SDK 확인 (https://en.wikipedia.org/wiki/CUDA)

GPUs supported

Supported CUDA level of GPU and card. See also at Nvidia:

  • CUDA SDK 1.0 support for compute capability 1.0 – 1.1 (Tesla)[21]
  • CUDA SDK 1.1 support for compute capability 1.0 – 1.1+x (Tesla)
  • CUDA SDK 2.0 support for compute capability 1.0 – 1.1+x (Tesla)
  • CUDA SDK 2.1 – 2.3.1 support for compute capability 1.0 – 1.3 (Tesla)[22][23][24][25]
  • CUDA SDK 3.0 – 3.1 support for compute capability 1.0 – 2.0 (Tesla, Fermi)[26][27]
  • CUDA SDK 3.2 support for compute capability 1.0 – 2.1 (Tesla, Fermi)[28]
  • CUDA SDK 4.0 – 4.2 support for compute capability 1.0 – 2.1+x (Tesla, Fermi, more?).
  • CUDA SDK 5.0 – 5.5 support for compute capability 1.0 – 3.5 (Tesla, Fermi, Kepler).
  • CUDA SDK 6.0 support for compute capability 1.0 – 3.5 (Tesla, Fermi, Kepler).
  • CUDA SDK 6.5 support for compute capability 1.1 – 5.x (Tesla, Fermi, Kepler, Maxwell). Last version with support for compute capability 1.x (Tesla)
  • CUDA SDK 7.0 – 7.5 support for compute capability 2.0 – 5.x (Fermi, Kepler, Maxwell)
  • CUDA SDK 8.0 support for compute capability 2.0 – 6.x (Fermi, Kepler, Maxwell, Pascal). Last version with support for compute capability 2.x (Fermi) (Pascal GTX 1070Ti Not Support)
  • CUDA SDK 9.0 – 9.2 support for compute capability 3.0 – 7.2 (Kepler, Maxwell, Pascal, Volta) (Pascal GTX 1070Ti Not Support CUDA SDK 9.0 and support CUDA SDK 9.2)
  • CUDA SDK 10.0 – 10.2 support for compute capability 3.0 – 7.5 (Kepler, Maxwell, Pascal, Volta, Turing). Last version with support for compute capability 3.x (Kepler). 10.2 is the last official release for macOS, as support will not be available for macOS in newer releases.
  • CUDA SDK 11.0 – support for compute capability 5.2 - 8.0 (Maxwell, Pascal, Volta, Turing, Ampere)

4. CUDA 설치

- 설치후 환경 변수 추가 확인

- 환경변수 등록은 자동으로 됩니다. 아래와 같이 잘 등록이 자동으로 되었나 확인 해 봅시다.

(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1)

- 위 파란색 경로를 복사 해 둡니다. 그리고 아래와 같이 진행 합니다.

1) '시작 - 찾기 - 시스템 환경 변수 편집'에 들어갑니다.

2) 확인

아래 CUDA_PATH와 CUDA_PATH_V10_1에 등록이 되어 있는 걸 확인 할 수 있습니다.


4. cuDNN 설치

 CUDA 버전에 맞게 알맞는 cuDNN을 선택 할 수 있습니다.

- 다운로드가 완료 되면 아래와 같이 다운로드 된 경로에 'cuda'라은 파일이 생기고 하위 폴더에 아래와 같이 파일이 생성 되어 있을 겁니다.

 

3) 경로 이동

- 위 쿠다 파일의 bin, include, lib 안에 있는 것들을 아까 4-3에서 설치 한 CUDA의 경로에 알맞게 각각 넣어 줍니다.

 

- 즉, C:\Users\KMG\Downloads\cudnn-10.1-windows10-x64-v7.6.4.38\cuda 안에 bin, include, lib-x64의 하위에 있는 cudnn64_7.dll, cudnn.h, cudnn.lib 이 세개를 알맞는 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1 의 bin, include, lib-x64 안으로 넣어야 합니다.

728x90

'자료' 카테고리의 다른 글

C#에서 YOLO v3 연동  (1) 2020.07.28
OpenCV 설치  (0) 2020.07.28
YOLO V3 cfg 파일 설정  (0) 2020.07.28
YOLO V3 윈도우환경 설치 -2  (0) 2020.07.26
YOLO-V3윈도우 환경 설치  (0) 2020.07.26