본문 바로가기

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' 카테고리의 다른 글

[Docker] failed to Lchown  (0) 2023.06.07
[Docker] nvidia-docker의 Dockerfile  (0) 2023.04.06
[Docker] Docker run 옵션 추가하기  (0) 2021.12.03
[Docker] link 후 jupyter notebook, -p(port) 옵션  (0) 2021.11.29
[Docker] nvidia-docker  (0) 2021.11.29