From 12ccb0b3c3ee1e7596b6126c2130150c1b31d272 Mon Sep 17 00:00:00 2001 From: Rob Ballantyne Date: Fri, 14 Jun 2024 19:03:07 +0100 Subject: [PATCH] update links. Ensure ROCm 6.0 --- .github/workflows/docker-build.yml | 4 ++-- README.md | 16 ++++++++-------- 2 files changed, 10 insertions(+), 10 deletions(-) diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml index b5cfd79..aebebb8 100644 --- a/.github/workflows/docker-build.yml +++ b/.github/workflows/docker-build.yml @@ -172,14 +172,14 @@ jobs: - "2.3.1" rocm: - "5.7" - - "6.0.2" + - "6.0" level: - "runtime" exclude: - pytorch: "2.3.1" rocm: "5.7" - pytorch: "2.2.1" - rocm: "6.0.2" + rocm: "6.0" steps: - diff --git a/README.md b/README.md index 6a448cd..9d9b9c0 100644 --- a/README.md +++ b/README.md @@ -8,7 +8,7 @@ This image provides a great starting point for python development when used stan ## 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/template) as straightforward and user friendly as possible. +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. @@ -20,25 +20,25 @@ The `:latest` tag points to `:latest-cuda` Tags follow these patterns: ##### _CUDA_ -- `:[pytorch-version]-py[python-version]-cuda-[x.x.x]-base-[ubuntu-version]` +- `:[pytorch-version]-py[python-version]-v2-cuda-[x.x.x]-base-[ubuntu-version]` -- `:latest-cuda` → `:2.2.1-py3.10-cuda-11.8.0-base-22.04` +- `:latest-cuda` → `:2.3.1-py3.10-v2-cuda-11.8.0-base-22.04` ##### _ROCm_ -- `:[pytorch-version]-py[python-version]-rocm-[x.x.x]-runtime-[ubuntu-version]` +- `:[pytorch-version]-py[python-version]-v2-rocm-[x.x.x]-runtime-[ubuntu-version]` -- `:latest-rocm` → `:2.2.1-py3.10-rocm-5.7-runtime-22.04` +- `:latest-rocm` → `:2.3.1-py3.10-v2-rocm-6.0-runtime-22.04` ##### _CPU_ -- `:[pytorch-version]-py[python-version]-ubuntu-[ubuntu-version]` +- `:[pytorch-version]-py[python-version]-v2-cpu-[ubuntu-version]` -- `:latest-cpu` → `:2.2.1-py3.10-cpu-22.04` +- `: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.2.1` `2.1.2` +Supported Pytorch versions: `2.3.1` `2.2.1` Supported Platforms: `NVIDIA CUDA`, `AMD ROCm`, `CPU`