We provide GPU-enabled docker images including Keras, TensorFlow, CNTK, MXNET and Theano.
-
Updated
Sep 8, 2019 - Dockerfile
#
docker-gpu
Here are 6 public repositories matching this topic...
A repository of docker-compose yaml files for all kinds of deployments
-
Updated
Jul 9, 2024 - Shell
Dockerfiles and manual for easy build of docker image with CUDA10.X and cuDNN7.6 to run TensorFlow/PyTorch on the nvidia GPU in docker-container.
docker dockerfile tensorflow gpu docker-image cuda torch pytorch nvidia nvidia-docker cudnn tensorflow-examples tensorflow-gpu docker-gpu docker-nvidia pytorch-gpu gpu-docker tensorflow-gpu-docker
-
Updated
Apr 14, 2021 - Roff
A collection of Docker compose scripts centered around use in a home lab for learning IT technologies.
-
Updated
Jan 4, 2026 - Shell
An Ansible role to deploy a docker environment on specific linux systems (with some extras like ctop, docker-compose, GPU support in docker for Nvidia graphic cards)
docker ansible docker-compose ansible-role nvidia-docker datadrivers datadrivers-gmbh ctop gpu-support docker-gpu
-
Updated
Jul 2, 2018 - Python
Reproducible Docker + NVIDIA GPU setup on Ubuntu for CUDA and PyTorch workloads. Covers NVIDIA Container Toolkit, GPU passthrough, explicit CUDA verification, and debuggable container workflows. Designed for correctness, reproducibility, and ML infrastructure work.
linux docker machine-learning gpu cuda pytorch nvidia containerization reproducible-builds ml-infrastructure docker-gpu nvidia-container-toolkit gpu-containers
-
Updated
Jan 8, 2026 - Shell
Improve this page
Add a description, image, and links to the docker-gpu topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the docker-gpu topic, visit your repo's landing page and select "manage topics."