반응형
nVidia
-
docker --runtime=nvidia default로 바꾸기 (GPU 사용 default)개발 환경/Docker 2020. 2. 26. 13:24
docker container에서 GPU를 사용하기 위해 docker의 runtime을 nvidia로 설정해줘야한다. 하지만 Pycharm과 같은 IDE에서 docker를 사용할 때는 매우 번거롭다. 그러므로 docker 차원에서 runtime을 nvidia로 default 설정해주자. 1. nvidia-smi test docker run --rm ubuntu nvidia-smi output: nvidia-smi: command not found --runtime=nvidia docker run --rm --runtime=nvidia ubuntu nvidia-smi output: Wed Feb 26 04:19:07 2020 +--------------------------------------------..