您的当前位置:首页正文

可使用 GPU 的 DOCKER 容器

来源:花图问答

可使用 GPU 的 DOCKER 容器

docker-logo.png

在 GPU 加速的数据中心内轻松部署应用程序
容器将应用程序封装到隔离的虚拟环境中,以简化数据中心的部署。通过将所有应用程序依赖项 (例如二进制文件和库) 都包括在内,应用程序容器能在任何数据中心环境中无缝地运行。
Docker 是领先的容器平台,它现在可用于容器化 GPU 加速的应用程序。这意味着无需进行任何修改即可轻松容器化和隔离加速的应用程序,并将其部署到任何受支持的、可使用 GPU 的基础架构上。 管理和监控加速的数据中心将变得空前容易。

gpu-enabled-docker-containers.png

重要好处
可以将旧的加速计算应用程序容器化,并部署在较新的系统、内部环境或云中。
可以将特定的 GPU 资源分配给容器,以获得更好的隔离效果和性能。
可以轻松地跨不同的环境共享应用程序、协同工作和测试应用程序。

# Install nvidia-docker and nvidia-docker-plugin
wget -P /tmp 
sudo dpkg -i /tmp/nvidia-docker*.deb && rm /tmp/nvidia-docker*.deb

# Test nvidia-smi
nvidia-docker run --rm nvidia/cuda nvidia-smi
# Install nvidia-docker and nvidia-docker-plugin
wget -P /tmp 
sudo rpm -i /tmp/nvidia-docker*.rpm && rm /tmp/nvidia-docker*.rpm
sudo systemctl start nvidia-docker

# Test nvidia-smi
nvidia-docker run --rm nvidia/cuda nvidia-smi
# Install nvidia-docker and nvidia-docker-plugin
wget -P /tmp  tar --strip-components=1 -C /usr/bin -xvf /tmp/nvidia-docker*.tar.xz && rm /tmp/nvidia-docker*.tar.xz
# Run nvidia-docker-plugin
sudo -b nohup nvidia-docker-plugin > /tmp/nvidia-docker.log

# Test nvidia-smi
nvidia-docker run --rm nvidia/cuda nvidia-smi