티스토리 뷰

728x90

 

 

컨테이너를 실행할 때 nvidia gpu를 사용할 수 있도록 --gpus 옵션을 사용하여 지정할 수 있다.

--gpus '"device={...}"' 

이런 형식으로 설정할 수 있고, 개인 서버를 할당 받아서 하나씩 사용하다보니 항상 device 번호 지정된 것 하나만 불러왔었는데 여러 개 지정해도 된다는 사실을 알았다.

 

그리고 내가 만든 컨테이너 내부에서 gpu 정보를 확인하고 싶어졌다.

 

도커 생성한 컨테이너에 연결된 gpu 확인

docker exec -it {컨테이너 이름} nvidia-smi

 

위 명령을 사용하면 컨테이너 내부에서 사용 가능한 gpu에 대한 상세한 정보가 출력되고, gpu 번호, 이름, 드라이버 정보, 메모리 사용 등을 확인할 수 있다.

728x90
댓글