add missing ocl package. Bump latest to 2.2.1

.github/workflows/docker-build.yml

@@ -8,9 +8,9 @@ on:
 env:
   UBUNTU_VERSION: 22.04
   BUILDX_NO_DEFAULT_ATTESTATIONS: 1
-  LATEST_CUDA: "2.2.0-py3.12-cuda-12.2.0-base-22.04"
-  LATEST_ROCM: "2.2.0-py3.12-rocm-6.0.2-runtime-22.04"
-  LATEST_CPU: "2.2.0-py3.12-cpu-22.04"
+  LATEST_CUDA: "2.2.1-py3.12-cuda-12.2.0-base-22.04"
+  LATEST_ROCM: "2.2.1-py3.12-rocm-6.0.2-runtime-22.04"
+  LATEST_CPU: "2.2.1-py3.12-cpu-22.04"
 
 jobs:
   cpu-base:
@@ -23,12 +23,9 @@ jobs:
           - "3.11"
           - "3.12"
         pytorch:
-          - "2.1.1"
           - "2.1.2"
-          - "2.2.0"
+          - "2.2.1"
         exclude:
-          - python: "3.12"
-            pytorch: "2.1.1"
           - python: "3.12"
             pytorch: "2.1.2"
     steps:
@@ -98,9 +95,8 @@ jobs:
           - "3.11"
           - "3.12"
         pytorch:
-          - "2.1.1"
           - "2.1.2"
-          - "2.2.0"
+          - "2.2.1"
         cuda:
           - "11.8.0"
           - "12.1.0"
@@ -109,8 +105,6 @@ jobs:
           - "runtime"
           - "devel"
         exclude:
-          - python: "3.12"
-            pytorch: "2.1.1"
           - python: "3.12"
             pytorch: "2.1.2"
 
@@ -181,14 +175,14 @@ jobs:
           - "3.12"
         pytorch:
           - "2.1.2"
-          - "2.2.0"
+          - "2.2.1"
         rocm:
           - "5.6"
           - "5.7"
         level:
           - "runtime"
         exclude:
-          - pytorch: "2.2.0"
+          - pytorch: "2.2.1"
             rocm: "5.6"
           - pytorch: "2.1.2"
             rocm: "5.7"

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.2.0-py3.12-cuda-12.1.0-base-22.04`
+- `:latest-cuda` → `:2.2.1-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.2.0-py3.12-rocm-5.7-runtime-22.04`
+- `:latest-rocm` → `:2.2.1-py3.12-rocm-5.7-runtime-22.04`
 
 ##### _CPU_
 - `:[pytorch-version]-py[python-version]-ubuntu-[ubuntu-version]`
 
-- `:latest-cpu` → `:2.2.0-py3.12-cpu-22.04` 
+- `:latest-cpu` → `:2.2.1-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.12`, `3.11`, `3.10`
 
-Supported Pytorch versions: `2.2.0` `2.1.2`
+Supported Pytorch versions: `2.2.1` `2.1.2`
 
 Supported Platforms: `NVIDIA CUDA`, `AMD ROCm`, `CPU`
 

build/COPY_ROOT/opt/ai-dock/bin/build/layer0/common.sh

@@ -18,4 +18,5 @@ printf "export MAMBA_INSTALL=\"%s\"\n" "${MAMBA_INSTALL}" >> /opt/ai-dock/etc/en
 
 $MAMBA_INSTALL -n $MAMBA_DEFAULT_ENV \
     ffmpeg="$ffmpeg_version" \
-    sox=14.4.2
+    sox=14.4.2 \
+    ocl-icd-system