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

PyTorch 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/pytorch

ai cuda docker jupyter machine-learning python pytorch rocm runpod syncthing vast

Last synced: 2 months ago
JSON representation

PyTorch 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/pytorch
• Owner: ai-dock
• License: other
• Created: 2023-08-04T15:20:08.000Z (over 1 year ago)
• Default Branch: main
• Last Pushed: 2024-06-14T18:03:13.000Z (8 months ago)
• Last Synced: 2024-06-15T18:00:55.939Z (8 months ago)
• Topics: ai, cuda, docker, jupyter, machine-learning, python, pytorch, rocm, runpod, syncthing, vast
• Language: Shell
• Homepage:
• Size: 84 KB
• Stars: 4
• Watchers: 4
• Forks: 2
• 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/pytorch/actions/workflows/docker-build.yml/badge.svg)](https://github.com/ai-dock/pytorch/actions/workflows/docker-build.yml)

# AI-Dock + PyTorch

Run python in a cloud-first AI-Dock container with PyTorch pre-installed.

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_

- `:[pytorch-version]-py[python-version]-v2-cuda-[x.x.x]-base-[ubuntu-version]`

- `:latest-cuda` → `:2.3.1-py3.10-v2-cuda-11.8.0-base-22.04`

##### _ROCm_

- `:[pytorch-version]-py[python-version]-v2-rocm-[x.x.x]-runtime-[ubuntu-version]`

- `:latest-rocm` → `:2.3.1-py3.10-v2-rocm-6.0-runtime-22.04`

##### _CPU_

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

- `:latest-cpu` → `:2.3.1-py3.10-v2-cpu-22.04` 

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

Supported Python versions: `3.12`, `3.11`, `3.10`

Supported Pytorch versions: `2.3.1` `2.2.1`

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

## Pre-Configured Templates

**Vast.​ai**

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

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

**Runpod.​io**

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

>[!NOTE]  

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

---

_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_