참고 사이트
nvidia.github.io/nvidia-docker/
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
댓글