mirror of
https://github.com/superlinear-ai/python-gpu.git
synced 2025-03-26 03:37:17 +03:00
32 lines
1.0 KiB
Markdown
32 lines
1.0 KiB
Markdown
[](https://hub.docker.com/r/superlinear/python-gpu)
|
|
|
|
# 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](https://hub.docker.com/r/superlinear/python-gpu/tags) for a full list.
|
|
|
|
### Running the image
|
|
|
|
```sh
|
|
docker run -it --rm superlinear/python-gpu:3.11-cuda11.8 /bin/bash
|
|
```
|
|
|
|
### Extending the image
|
|
|
|
```Dockerfile
|
|
FROM superlinear/python-gpu:3.11-cuda11.8
|
|
|
|
...
|
|
```
|