Skip to content

Commit 75783d8

Browse files
authored
docs: add README
1 parent 80f9277 commit 75783d8

File tree

1 file changed

+31
-0
lines changed

1 file changed

+31
-0
lines changed

README.md

+31
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,31 @@
1+
[![Open in Docker Hub](https://img.shields.io/static/v1?label=Docker%20Hub&message=Open&color=blue&logo=dockerhub)](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
23+
```
24+
25+
### Extending the image
26+
27+
```Dockerfile
28+
FROM radixai/python-gpu:3.11-cuda11.8
29+
30+
...
31+
```

0 commit comments

Comments
 (0)