Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/ai-dock/python

Python docker images for use in GPU cloud and local environments. Includes AI-Dock base for authentication and improved user experience.
https://github.com/ai-dock/python

ai cuda docker machine-learning python rocm runpod vast

Last synced: 2 months ago
JSON representation

Python docker images for use in GPU cloud and local environments. Includes AI-Dock base for authentication and improved user experience.

Host: GitHub
URL: https://github.com/ai-dock/python
Owner: ai-dock
License: other
Created: 2023-08-04T13:45:56.000Z (over 1 year ago)
Default Branch: main
Last Pushed: 2024-08-07T07:08:33.000Z (6 months ago)
Last Synced: 2024-08-07T10:32:33.451Z (6 months ago)
Topics: ai, cuda, docker, machine-learning, python, rocm, runpod, vast
Language: Shell
Homepage:
Size: 93.8 KB
Stars: 3
Watchers: 2
Forks: 0
Open Issues: 0
Metadata Files:
- Readme: README.md
- Funding: .github/FUNDING.yml
- License: LICENSE.md

Awesome Lists containing this project

README

        [![Docker Build](https://github.com/ai-dock/python/actions/workflows/docker-build.yml/badge.svg)](https://github.com/ai-dock/python/actions/workflows/docker-build.yml)

# AI-Dock + Python

Run python in a cloud-first AI-Dock container. Nothing is added to the installed python environment(s) - You'll have python & pip.

This image provides a great starting point for python development when used standalone but its also a solid foundation for extending upon.

## Documentation

All AI-Dock containers share a common base which is designed to make running on cloud services such as [vast.ai](https://link.ai-dock.org/vast.ai) and [runpod.io](https://link.ai-dock.org/runpod.io) as straightforward and user friendly as possible.

Common features and options are documented in the [base wiki](https://github.com/ai-dock/base-image/wiki) but any additional features unique to this image will be detailed below.

## Version Tags

The `:latest` tag points to `:latest-cuda`

Tags follow these patterns:

##### _CUDA_

- `:[python-version]-v2-cuda-[x.x.x]-[base|runtime|devel]-[ubuntu-version]`

- `:latest-cuda` → `:3.10-v2-cuda-11.8.0-cudnn8-runtime-22.04`

##### _ROCm_

- `:[python-version]-v2-rocm-[x.x.x]-[core|runtime]-[ubuntu-version]`

- `:latest-rocm` → `:3.10-v2-rocm-6.0-runtime-22.04`

ROCm builds are experimental. Please give feedback.

##### _CPU_

- `:[python-version]-v2-cpu-[ubuntu-version]`

- `:latest-cpu` → `:3.10-v2-cpu-22.04`

Browse [here](https://github.com/ai-dock/python/pkgs/container/python) for an image suitable for your target environment.

Supported Python versions: `3.10`

Supported Platforms: `NVIDIA CUDA`, `AMD ROCm`, `CPU`

>[!NOTE]  

>Recent builds include `v2` in their image tag.  These images use `venv` rather than `micromamba` for environment management.

## Pre-Configured Templates

**Vast.ai**

[python:latest-cuda](https://link.ai-dock.org/template-vast-python) (CUDA)

[python:latest-rocm](https://link.ai-dock.org/template-vast-python-rocm) (ROCm)

---

**Runpod.io**

[python:latest](https://link.ai-dock.org/template-runpod-python)

---

>[!NOTE]  

>These templates are configured to use the `latest` tag but you are free to change to any of the available Python CUDA tags listed [here](https://github.com/ai-dock/python/pkgs/container/python)

---

_The author ([@robballantyne](https://github.com/robballantyne)) may be compensated if you sign up to services linked in this document. Testing multiple variants of GPU images in many different environments is both costly and time-consuming; This along with [sponsorships](https://github.com/sponsors/ai-dock) helps to offset costs and further the development of the project_