Tag Archives: cuDNN

#Docker with #Nvidia runtime for #Deep #Learning

Just recently got to know that it’s possible to leverage CUDA and cuDNN inside Docker.

Basically what is needed is corresponding Docker image that can be found here  installation of Nvidia Docker runtime and to configure Docker daemon to use nvidia as a default runtime in /etc/docker/daemon.json. as it was described here and bellow there is an example of my config:

      "default-runtime": "nvidia",
      "runtimes": {
          "nvidia": {
              "path": "/usr/bin/nvidia-container-runtime",
              "runtimeArgs": []