神刀安全网

Containerized Nvidia Docker Image for GPU Applications

NVIDIA Docker

This repository includes utilities to build and run NVIDIA Docker images.

Containerized Nvidia Docker Image for GPU Applications

Example of how CUDA integrates with Docker

Documentation

The full documentation is available on therepository wiki.

A good place to start is to understandwhy NVIDIA Docker is needed in the first place.

Quick start

Assuming the NVIDIA drivers and Docker are properly installed (seeinstallation), there are two ways to install NVIDIA Docker:

Plugin install (Recommended)

Ubuntu distributions

# Install nvidia-docker and nvidia-docker-plugin wget -P /tmp https://github.com/NVIDIA/nvidia-docker/releases/download/v1.0.0-rc/nvidia-docker_1.0.0.rc-1_amd64.deb sudo dpkg -i /tmp/nvidia-docker_1.0.0.rc-1_amd64.deb && rm /tmp/nvidia-docker*.deb  # Test nvidia-smi nvidia-docker run --rm nvidia/cuda nvidia-smi

Other distributions

# Install nvidia-docker and nvidia-docker-plugin wget -P /tmp https://github.com/NVIDIA/nvidia-docker/releases/download/v1.0.0-rc/nvidia-docker_1.0.0.rc_amd64.tar.xz sudo tar --strip-components=1 -C /usr/bin -xvf /tmp/nvidia-docker_1.0.0.rc_amd64.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

Standalone install

# Install nvidia-docker and nvidia-docker-plugin wget -P /tmp https://github.com/NVIDIA/nvidia-docker/releases/download/v1.0.0-rc/nvidia-docker_1.0.0.rc_amd64.tar.xz sudo tar --strip-components=1 -C /usr/bin -xvf /tmp/nvidia-docker_1.0.0.rc_amd64.tar.xz && rm /tmp/nvidia-docker*.tar.xz  # One-time setup sudo nvidia-docker volume setup  # Test nvidia-smi nvidia-docker run --rm nvidia/cuda nvidia-smi

Issues and Contributing

A signed copy of the Contributor License Agreement needs to be provided to digits@nvidia.com before any change can be accepted.

  • Please let us know byfiling a new issue
  • You can contribute by opening apull request

转载本站任何文章请注明:转载至神刀安全网,谢谢神刀安全网 » Containerized Nvidia Docker Image for GPU Applications

分享到:更多 ()

评论 抢沙发

  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址