https://github.com/superlinear-ai/python-gpu

🐳 Python GPU adds a minimal install of CUDA and cuDNN on top of the official python:3.x-slim base image
https://github.com/superlinear-ai/python-gpu

cuda cudnn docker docker-image python

Last synced: 2 months ago
JSON representation

🐳 Python GPU adds a minimal install of CUDA and cuDNN on top of the official python:3.x-slim base image

• Host: GitHub
• URL: https://github.com/superlinear-ai/python-gpu
• Owner: superlinear-ai
• Created: 2023-07-17T12:43:13.000Z (over 1 year ago)
• Default Branch: main
• Last Pushed: 2024-09-14T18:06:15.000Z (4 months ago)
• Last Synced: 2024-09-15T03:37:33.922Z (4 months ago)
• Topics: cuda, cudnn, docker, docker-image, python
• Language: Dockerfile
• Homepage:
• Size: 43.9 KB
• Stars: 6
• Watchers: 2
• Forks: 1
• Open Issues: 1
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
[![Open in Docker Hub](https://img.shields.io/static/v1?label=Docker%20Hub&message=Open&color=blue&logo=dockerhub)](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}, 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

...

```