Docker Nvidia-Docker GPU 사용 (이전 완료)

이전에는 nvidia-docker로 따로 사용하던게 docker run 명령어의 옵션 –gpus {GPU NUMBER}으로 통합되었다.

예전

nvidia-docker run nvidia/cuda:11.0-base …

 

현재

docker run --gpus 1 nvidia/cuda:11.0-base

 

{GPU Number}은 GPU몇번을 사용하겠다는 의미다.

docker run --gpus 0 nvidia/cuda:11.0-base    
docker run --gpus all nvidia/cuda:11.0-base

 

0 : 0번째 GPU만 사용
all : 모든 GPU를 사용