Skip to content

Commit

Permalink
docs: add README
Browse files Browse the repository at this point in the history
  • Loading branch information
lsorber authored Sep 12, 2023
1 parent 80f9277 commit 75783d8
Showing 1 changed file with 31 additions and 0 deletions.
31 changes: 31 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,31 @@
[![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)

# 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} and CUDA {11.8}
- 📦 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 `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.

### Running the image

```sh
docker run -it --rm radixai/python-gpu:3.11-cuda11.8 /bin/bash
```

### Extending the image

```Dockerfile
FROM radixai/python-gpu:3.11-cuda11.8

...
```

0 comments on commit 75783d8

Please sign in to comment.