docker gpu

[Docker] Docker container에서 GPU 사용하기

먼저 터미널에 nvidia-smi를 통해서 GPU를 사용하고 있는지 확인 쓰니는 이미 container에 GPU를 지정해줘서 colmap을 사용하고 있다. GPU 옵션 1. GPU 모두 사용 $ docker run --gpus all -it --name [컨테이너이름] [이미지이름 or 이미지ID] # 1번 GPU만 사용하기 2. GPU 지정해서 사용하기 # 1개 지정해서 사용 $ docker run --gpus '"device=1"' -it --name [컨테이너이름] [이미지이름 or 이미지ID] # 1번 GPU만 사용하기 # 여러개 지정해서 사용 $ docker run --gpus '"device=0,2"' -it --name [컨테이너이름] [이미지이름 or 이미지ID] # 0번, 2번 GPU만 ..