From 50e1b25d8d1288ca6d6da066aff9a76e8c554bf9 Mon Sep 17 00:00:00 2001 From: Rob Ballantyne Date: Thu, 8 Feb 2024 00:10:39 +0000 Subject: [PATCH] Fix build targets --- .github/workflows/docker-build.yml | 29 ++++++++++++++++++++--------- README.md | 10 +++++----- 2 files changed, 25 insertions(+), 14 deletions(-) diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml index eac7007..57b0942 100644 --- a/.github/workflows/docker-build.yml +++ b/.github/workflows/docker-build.yml @@ -8,9 +8,9 @@ on: env: UBUNTU_VERSION: 22.04 BUILDX_NO_DEFAULT_ATTESTATIONS: 1 - LATEST_CUDA: "2.1.1-py3.11-cuda-12.2.0-base-22.04" - LATEST_ROCM: "2.1.1-py3.11-rocm-5.6-runtime-22.04" - LATEST_CPU: "2.1.1-py3.11-cpu-22.04" + LATEST_CUDA: "2.2.0-py3.12-cuda-12.2.0-base-22.04" + LATEST_ROCM: "2.1.2-py3.11-rocm-5.6-runtime-22.04" + LATEST_CPU: "2.2.0-py3.12-cpu-22.04" jobs: cpu-base: @@ -21,10 +21,16 @@ jobs: python: - "3.10" - "3.11" + - "3.12" pytorch: - - "2.0.1" - - "2.1.0" - "2.1.1" + - "2.1.2" + - "2.2.0" + exclude: + - python: "3.12" + pytorch: "2.1.1" + - python: "3.12" + pytorch: "2.1.2" steps: - name: Free Space @@ -82,17 +88,23 @@ jobs: python: - "3.10" - "3.11" + - "3.12" pytorch: - - "2.1.0" - "2.1.1" + - "2.1.2" + - "2.2.0" cuda: - "11.8.0" - "12.1.0" - - "12.2.0" level: - "base" - "runtime" - "devel" + exclude: + - python: "3.12" + pytorch: "2.1.1" + - python: "3.12" + pytorch: "2.1.2" steps: - @@ -151,8 +163,7 @@ jobs: - "3.10" - "3.11" pytorch: - - "2.1.0" - - "2.1.1" + - "2.1.2" rocm: - "5.6" level: diff --git a/README.md b/README.md index 6788b8d..5c51336 100644 --- a/README.md +++ b/README.md @@ -32,25 +32,25 @@ Tags follow these patterns: ##### _CUDA_ - `:[pytorch-version]-py[python-version]-cuda-[x.x.x]-base-[ubuntu-version]` -- `:latest-cuda` → `:2.1.1-py3.11-cuda-12.1.0-base-22.04` +- `:latest-cuda` → `:2.2.0-py3.12-cuda-12.1.0-base-22.04` ##### _ROCm_ - `:[pytorch-version]-py[python-version]-rocm-[x.x.x]-runtime-[ubuntu-version]` -- `:latest-rocm` → `:2.1.1-py3.11-rocm-5.6-runtime-22.04` +- `:latest-rocm` → `:2.1.2-py3.12-rocm-5.6-runtime-22.04` ##### _CPU_ - `:[pytorch-version]-py[python-version]-ubuntu-[ubuntu-version]` -- `:latest-cpu` → `:2.1.1-py3.11-cpu-22.04` +- `:latest-cpu` → `:2.2.0-py3.12-cpu-22.04` Browse [here](https://github.com/ai-dock/pytorch/pkgs/container/pytorch) for an image suitable for your target environment. You can also self-build from source by editing `.env` and running `docker compose build`. -Supported Python versions: `3.11`, `3.10` +Supported Python versions: `3.12`, `3.11`, `3.10` -Supported Pytorch versions: `2.1.1` `2.1.0` `2.0.1` +Supported Pytorch versions: `2.2.0` `2.1.2` Supported Platforms: `NVIDIA CUDA`, `AMD ROCm`, `CPU`