티스토리 뷰

728x90

Yolov3, Yolov4를 사용하여 작품을 만들 생각이라면 먼저,

사용 중인 노트북 혹은 컴퓨터에 NVIDIA 그래픽 드라이버가 장착되어있는지 확인해야합니다.

테스트를 진행할 때도 GPU를 사용해야하기 때문에 꼭 꼭 필요합니다.

nvidia 드라이버가 없는 노트북으로 왜 추가 드라이버가 안뜨지 ?? ㅠㅠㅠㅠ 하면서 2달을 허비하는 불상사가 저희 말고는 없길 바라며.. ! ! !

만약 드라이버가 없는 노트북이라도 렌탈하면 되니 걱정하지 마세요 :)

 

라즈베리파이4에서도 해보려고 했으나, 성능 문제로 조금만 무리하면 killed 되는 오류가 있었습니다. 

그래서 저희는 core i7, NVIDIA 840M GTX 가 장착된 노트북을 렌탈 후 진행하였습니다.

학습시키기에는 턱없이 부족한 성능이므로 학습은 Portainer를 통해 Docker에 접근하여 진행하였습니다.

 

portainer을 통해 학습은 시킬 수 있었으나, GUI가 지원되지 않아

labeling, 예제 테스트를 진행 할 수 없어 ssd에 ubuntu 설치 후 엔비디아 드라이버가 장착되어 있는 렌탈한 노트북에 연결해서 사용했습니다.

 

  • Ubuntu 16.04LTS (cuda 버전때문에) 

        - nvidia-384 (그래픽 드라이버)

        - CUDA 9.0

        - Cudnn 

  • Opencv-4.0.2 ver.
  • Yolov3
  • 테스트용 - Nvidia geforce GTX 840M
  • 학습용 - NVIDIA geforce RTX 2080 x 2

 

 


1. Portainer 사용

Portainer를 이용한 Docker 사용하기

1) image 생성하기

https://hub.docker.com/

Docker Hub

Docker Certified:Trusted & Supported Products Certified Containers provide ISV apps available as containers. Certified Plugins for networking and volumes in containers. Certified Infrastructure delivers an optimized and validated Docker platform for enterp

hub.docker.com

 

 

 

 

도커이미지 공식 홈페이지에 들어가서 각자 원하는 이미지를 다운받아줍니다.

다운받은 후 portainer에서 이미지를 pull 해줍니다.

 

2) 컨테이너 만들기

컨테이너 각자 원하는 방법으로 설정하고, 이미지를 아까 pull한 이미지로 설정해줍니다.

 


 

반응형