mirror of
https://github.com/superlinear-ai/python-gpu.git
synced 2025-03-26 03:37:17 +03:00
dependabot/github_actions/ci-dependencies-bf940bc907
Bumps the ci-dependencies group with 1 update: [docker/build-push-action](https://github.com/docker/build-push-action). Updates `docker/build-push-action` from 5 to 6 - [Release notes](https://github.com/docker/build-push-action/releases) - [Commits](https://github.com/docker/build-push-action/compare/v5...v6) --- updated-dependencies: - dependency-name: docker/build-push-action dependency-type: direct:production update-type: version-update:semver-major dependency-group: ci-dependencies ... Signed-off-by: dependabot[bot] <support@github.com>
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-slimbase image - 🐍 Adds a single
micromambaexecutable to install CUDA and cuDNN - 🧬 Matrix build for Python {3.8, 3.9, 3.10, 3.11, 3.12}, CUDA {11.8}, and cuDNN {8.9}
- 📦 Multi-platform build for
linux/amd64andlinux/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
...
Languages
Dockerfile
100%