Update build targets

ai-dock · Feb 8, 2024 · e30b2a7 · e30b2a7
1 parent ede633f
commit e30b2a7
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 19 deletions.
diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml
@@ -8,8 +8,8 @@ on:
 env:
   UBUNTU_VERSION: 22.04
   BUILDX_NO_DEFAULT_ATTESTATIONS: 1
-  LATEST_CUDA: "3.12-cuda-12.2.0-runtime-22.04"
-  LATEST_ROCM: "3.12-rocm-5.6-runtime-22.04"
+  LATEST_CUDA: "3.12-cuda-12.3.1-runtime-22.04"
+  LATEST_ROCM: "3.12-rocm-6.0.2-runtime-22.04"
   LATEST_CPU: "3.12-cpu-22.04"
 
 jobs:
@@ -19,9 +19,6 @@ jobs:
       fail-fast: false
       matrix:
         python:
-          #- ["2.7", "python_27"]
-          #- ["3.8", "python_38"]
-          #- ["3.9", "python_39"]
           - ["3.10", "python_310"]
           - ["3.11", "python_311"]
           - ["3.12", "python_312"]
@@ -90,6 +87,7 @@ jobs:
           - 11.8.0
           - 12.1.0
           - 12.2.0
+          - 12.3.1
         level:
           - "base"
           - "runtime"
@@ -156,6 +154,8 @@ jobs:
         rocm:
           - 5.4.2
           - 5.6
+          - 5.7
+          - 6.0.2
         level:
           - "core"
           - "runtime"

diff --git a/README.md b/README.md
@@ -43,12 +43,12 @@ Tags follow these patterns:
 ##### _CUDA_
 - `:[python-version]-cuda-[x.x.x]-[base|runtime|devel]-[ubuntu-version]`
 
-- `:latest-cuda` &rarr; `:3.12-cuda-12.2.0-runtime-22.04`
+- `:latest-cuda` &rarr; `:3.12-cuda-12.3.1-runtime-22.04`
 
 ##### _ROCm_
 - `:[python-version]-rocm-[x.x.x]-[core|runtime]-[ubuntu-version]`
 
-- `:latest-rocm` &rarr; `:3.12-rocm-5.6-runtime-22.04`
+- `:latest-rocm` &rarr; `:3.12-rocm-6.0.2-runtime-22.04`
 
 ROCm builds are experimental. Please give feedback.
 

diff --git a/build/COPY_ROOT/opt/ai-dock/bin/build/layer0/clean.sh b/build/COPY_ROOT/opt/ai-dock/bin/build/layer0/clean.sh
@@ -4,13 +4,5 @@
 apt-get clean -y
 micromamba clean -ay
 
-# Remove build scripts
-scripts_dir="/opt/ai-dock/bin/build/"
-
-# Remove this layer's scripts
-rm -rf ${scripts_dir}layer0
-
-# Remove parent directory if this is the last build layer
-if [[ $(ls -l ${scripts_dir} | grep -c ^d) -eq 0 ]]; then
-    rm -rf ${scripts_dir}
-fi
+rm /etc/ld.so.cache
+ldconfig
diff --git a/build/COPY_ROOT/opt/ai-dock/bin/build/layer0/common.sh b/build/COPY_ROOT/opt/ai-dock/bin/build/layer0/common.sh
@@ -4,8 +4,6 @@ source /opt/ai-dock/etc/environment.sh
 
 build_common_main() {
     build_common_install_python
-    rm /etc/ld.so.cache
-    ldconfig
 }
 
 build_common_do_mamba_install() {