Fix build targets

.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:

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`