diff --git a/nvidia/nvidia-docker.sh b/nvidia/nvidia-docker.sh new file mode 100644 index 0000000..0056b66 --- /dev/null +++ b/nvidia/nvidia-docker.sh @@ -0,0 +1,14 @@ +#!/bin/bash +# thanks @88plug +curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \ + apt-key add - +distribution=$(. /etc/os-release;echo $ID$VERSION_ID) +curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | \ + tee /etc/apt/sources.list.d/nvidia-docker.list +apt-get update + +# Install nvidia-docker2 and reload the Docker daemon configuration +apt-get install -y nvidia-docker2 +pkill -SIGHUP dockerd + +reboot