You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
[](https://hub.docker.com/r/radixai/python-gpu)
2
+
3
+
# Python GPU
4
+
5
+
A minimal CUDA and cuDNN install on top of the official `python:3.x-slim` base image.
6
+
7
+
## 🎁 Features
8
+
9
+
- ✅ Starts from the official `python:3.x-slim` base image
10
+
- 🐍 Adds a single `micromamba` executable to install CUDA and cuDNN
11
+
- 🧬 Matrix build for Python {3.8, 3.9, 3.10, 3.11} and CUDA {11.8}
12
+
- 📦 Multi-platform build for `linux/amd64` and `linux/arm64`
13
+
- ✨ Image size is only 1.8GB
14
+
15
+
## ✨ Using
16
+
17
+
A matrix of tags are available that follow the format `radixai/python-gpu:$PYTHON_VERSION-cuda$CUDA_VERSION`, see the [Docker Hub repository](https://hub.docker.com/r/radixai/python-gpu/tags) for a full list.
18
+
19
+
### Running the image
20
+
21
+
```sh
22
+
docker run -it --rm radixai/python-gpu:3.11-cuda11.8 /bin/bash
0 commit comments