(Docker) nvidia-smi command not found

현상

내 local pc 자체에서 nvidia-smi 명령어를 치면 GPU 정보가 나옴.

엔비디아 드라이버가 깔린걸 인식.

그런데 도커 허브에서 받은 도커 환경 내부에서 nvidia-smi 명령어 치면 안 나옴.

실행커맨드

# Local Ubuntu PC Version 
docker run -v /home/경로/UCTransNet_code/UCTransNet:/workspace/TRANS_UNET  --name uctrans_cont  이미지명:latest #/bin/bash

원인

1. nvidia docker 안 깔림

• AI허브에는 깔았었는데 로컬 pc에 엔비디아 도커 설치 안한 거 까먹음.

2. 실행 커맨드

도커 run (컨테이너 만들기) 할 때

--ipc=host -it --gpus all 

같은 gpu 관련 옵션을 붙어야 도커 환경에서 gpu를 사용가능함.

• gpus all 옵션 -> 해당 device의 모든 GPU를 할당하여 도커 환경으로 진입함.

• docker hub에서 받은 이미지에 해당 옵션이 바로 실행되는건 도커허브의 이미지가 이미 nvidia pytorch .