[도커] 도커(Docker) 컨테이너에서 GPU 사용하는 방법

1. 모든 GPU를 사용하기

 

docker run --gpus all -i -t --name [컨테이너이름] [이미지이름 or 이미지ID]

 

2. 특정 GPU만 사용하기 

 

docker run --gpus device=1 -i -t --name [컨테이너이름] [이미지이름 or 이미지ID] # 1번 GPU만 사용하기 docker run --gpus device=0,2 -i -t --name [컨테이너이름] [이미지이름 or 이미지ID] # 0번, 2번 GPU만 사용하기

 

 

3. GPU를 사용하는지 알아보려면

nvidia-smi