[Docker] Tensorflow GPU Container 띄우기

# 명령어 기본
docker run -it --rm 
	--gpus [디바이스 할당] 
    -p [로컬포트넘버]:[컨테이너포트넘버] 
    --runtime=nvidia 
    -v "[로컬디렉토리]:[컨테이너 내 디렉토리]" 
    [도커이미지]:[태그]
    
1. -it	: i는 interactive하게 컨테이너와 상호작용, t는 tty를 사용하겠다는 뜻
2. --rm	: 컨테이너가 종료될 때 컨테이너와 관련된 리소스(파일 시스템, 볼륨)까지 깨끗이 제거
3. --gpus	: gpu device 할당 (all:모든 device할당  / 'device=1,2')
4. --runtime=nvidia	: nvidia-docker2 를 사용하는 경우
5. -p	: 연결할 호스트의 포트번호 : 컨테이너의 포트번호
6. -v	: 마운트할 호스트의 디렉토리 : 컨테이너의 디렉토리


# 필자 예시
$ docker run -it --rm --gpus all --runtime=nvidia -p 8989:8989 -p 6006:6006 -v "/home/jh3090/:/home/jh3090/" jh/tensorflow2:v1.2
$ docker run -it --rm --gpus '"device=1,2"' --runtime=nvidia -p 8989:8989 -p 6006:6006 -v "/home/jh3090/:/home/jh3090/" jh/tensorflow2:v1.2

 

GPU Device할당에 관련 된 포스팅은 추가적으로 업데이트 후 안내 하겠습니다!