본문 바로가기

IT/Docker

[Docker] nvidia-docker 설치 및 실행

 

 

Nvidia-docker 설치 및 설정 (Ubuntu20.04)

기준 사양RTX A6000 (Ampere Arch)Nvidia driver 470.xxCUDA 11.xUbuntu 20.04 LTShttps://docs.docker.com/engine/install/ubuntu/이전 버전 삭제https를 통해 reposito

velog.io

  • Setting up NVIDIA Container Toolkit (안정적인 레포지토리와 GPG Key 셋업)
$ distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
   && curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - \
   && curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list

 

 

[Docker] nvidia-docker2 설치하기 (Ubuntu 20.04)

1. Setting Up Docker (이미 도커가 설치 되어있는경우 이 과정을 Skip하자!!!!) $ curl https://get.docker.com | sh \ && sudo systemctl --now enable docker % Total % Received % Xferd Average Speed Time Time Time Current Dload Upload Total Spe

sseongju1.tistory.com

  • Docker에서 Cuda환경으로 시작하기

docker pull 로 당겨오기(nvidia/cuda:10.0-cudnn7-runtime-ubuntu18.04 으로 당겨오면 안됨!! nvcc 헤드가 없다
반드시 nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04 로 당겨와야함)

sudo docker pull nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04
  • docker로 컨테이너 환경에 접속
nvidia-docker run -it nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04 /bin/bash
  • cuda 및 드라이버 확인
nvdia-smi

 

 

#!/usr/bin/bash
docker run \
        -it \
        --rm \
        --gpus all \
        --shm-size=1g \
        -v /tmp/argus_socket:/tmp/argus_socket \
        -v /tmp/.X11-unix/:/tmp/.X11-unix \
        -v $HOME/tracker/:/tracker/ \
        nvidia/cuda:10.2-cudnn7-devel

'IT > Docker' 카테고리의 다른 글