Go to file
dependabot[bot] 84c98b509a ci(deps): bump docker/build-push-action in the ci-dependencies group
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>
2024-10-01 21:14:34 +00:00
2024-04-03 18:06:44 +02:00

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}, 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

...
Description
🐳 Python GPU adds a minimal install of CUDA and cuDNN on top of the official python:3.x-slim base image
Readme 122 KiB
Languages
Dockerfile 100%