본문 바로가기
OS/linux

🥰 centos nvidia-docker 설치

by Knowledge Store In Hyunsoft 2020. 11. 23.

참고 사이트 

nvidia.github.io/nvidia-docker/

 

Repository configuration

NVIDIA Docker Engine wrapper repository.

nvidia.github.io

1. yum(dnf) repo 추가

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.repo | \
  sudo tee /etc/yum.repos.d/nvidia-docker.repo

 

2. daemon.json backup 및 설치

cp /etc/docker/daemon.json /etc/docker/daemon.json.bak
sudo yum -y install nvidia-docker2

 

3.daemon.json 확인

cat /etc/docker/daemon.json

 

4. docker restart

systemctl restart docker

 

5. 확인

docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi
728x90

댓글