github

NVIDIA / nvidia-docker

  • воскресенье, 22 мая 2016 г. в 03:12:12
https://github.com/NVIDIA/nvidia-docker

Go
Build and run Docker containers leveraging NVIDIA GPUs



NVIDIA Docker

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

nvidia-gpu-docker

Example of how CUDA integrates with Docker

Documentation

The full documentation is available on the repository wiki.
A good place to start is to understand why NVIDIA Docker is needed in the first place.

Quick start

Assuming the NVIDIA drivers and Docker are properly installed (see installation), 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.