nvidia-docker安装

发表于 2024/03/29 18:49:12 [ai] 浏览次数：308

1.在宿主机安装显卡驱动,相应版本的cuda和cudnn

2.安装docker

3.安装nivdia-container-toolkit，官网地址:官网：https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html#installing-with-yum-or-dnf 给出yum安装步骤:

curl -s -L https://nvidia.github.io/libnvidia-container/stable/rpm/nvidia-container-toolkit.repo | \
  sudo tee /etc/yum.repos.d/nvidia-container-toolkit.repo

sudo yum-config-manager --enable nvidia-container-toolkit-experimental

sudo yum install -y nvidia-container-toolkit

sudo nvidia-ctk runtime configure --runtime=docker

重启：

sudo systemctl restart docker