Skip to content

🐳 Python GPU adds a minimal install of CUDA and cuDNN on top of the official python:3.x-slim base image

Notifications You must be signed in to change notification settings

superlinear-ai/python-gpu

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

48 Commits
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

Open in Docker Hub

Python GPU

A minimal CUDA and cuDNN install on top of the official python:3.x-slim base image.

🎁 Features

  • βœ… Starts from the official python:3.x-slim base image
  • 🐍 Adds a single micromamba executable to install CUDA and cuDNN
  • 🧬 Matrix build for Python {3.8, 3.9, 3.10, 3.11, 3.12, 3.13}, CUDA {11.8}, and cuDNN {8.9}
  • πŸ“¦ Multi-platform build for linux/amd64 and linux/arm64
  • 🐭 Image size is only 1.8GB

✨ Using

A matrix of tags are available that follow the format superlinear/python-gpu:$PYTHON_VERSION-cuda$CUDA_VERSION, see the Docker Hub repository for a full list.

Running the image

docker run -it --rm superlinear/python-gpu:3.11-cuda11.8 /bin/bash

Extending the image

FROM superlinear/python-gpu:3.11-cuda11.8

...

About

🐳 Python GPU adds a minimal install of CUDA and cuDNN on top of the official python:3.x-slim base image

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published