본문 바로가기
[Cloud]/심심풀이

[심심풀이] Tensoflow + nvidia_docer2 + cuda + GPU 관련자료를 찾아봤습니다.

by journes 2019. 3. 1.

 "Tensoflow + nvidia_docer2 + cuda + GPU " 찾아본 자료를 아래와 같이 간단히 정리해 봤습니다. :)

 - .   Nvidia_docker2 설치는 간단할것 같은데.. Tensoflow 와 연동은 어떻게 하는지 모르겠네요.

 

 

 

1. Nvidia GPU + CUDA

    - 현재 우리가 사용하는 기본적인 형태 ( CUDA는 병렬처리 용)

    - 

 


 

2.  Nvidia GPU + CUDA  + Nvidia_docker2 

   - "Nvidia_docker2"의 컨테이너를 여러개 띄워가용율을 높일수 있음


 

사용법은 아래와 같습니다.

  -  복수개의 GPU 선택이 가능합니다.

 

GPU 단일지정 예시) - NVIDIA_VISIBLE_DEVICES=0

 

$ docker run --rm -ti --runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=0 nvidia/cuda nvidia-smi

 

GPU 복수 지정 예시) -  NVIDIA_VISIBLE_DEVICES=1,2

$ docker run --rm -ti --runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=1,2 nvidia/cuda nvidia-smi

 

 

참고로,,, Nvidia_docker2  구성이 아래와 같습니다.

, Docker 환경에서,  여러개의 Container를 띄워서 사용할수 있습니다.

 

 


 

 

 

--------------------------------------------------------------------------------------------------------------------------------------------------------------

 

 

위와 같은내용에서 ,, 만약 아래와 같이  하나의 서버에 여러개의 GPU를 사용한다면가용성을 더욱 늘어날 것입니다.

 

 

2-1    Nvidia GPU (Multi)  + CUDA  + Nvidia_docker2  


 

 

마지막으로,   Tensorflow가 병렬처리를 지원해주면. .. 더욱 효과가 좋와지겠네요.. 

 

 

3. Tensoflow + nvidia_docer2 + cuda + GPU

 

 


 

 

사용 예시)