feat: pytorch 2.6.0, version pinning

.github/workflows/base.yml

@@ -5,6 +5,7 @@ on:
     paths:
       - "official-templates/base/**"
 
+
 jobs:
   build:
     runs-on: ubuntu-latest-public-m
@@ -42,9 +43,6 @@ jobs:
 
       - name: Set environment variables
         run: |
-          # Set environment variable to disable entitlements checks
-          echo "BUILDX_BAKE_ENTITLEMENTS_FS=0" >> $GITHUB_ENV
-
           # Only set RELEASE for development builds
           # For production, we'll use the default from docker-bake.hcl
           if [[ "${{ steps.build_type.outputs.is_production }}" != "true" ]]; then
@@ -56,8 +54,11 @@ jobs:
           fi
 
       - name: Build and push the images to Docker Hub
-        uses: docker/bake-action@v2
+        uses: docker/bake-action@v2 # latest: v6
+        env:
+          BUILDX_BAKE_ENTITLEMENTS_FS: 0
         with:
+          source: .
           files: ./official-templates/base/docker-bake.hcl
           push: true
           set: |

README.md

@@ -26,7 +26,7 @@ The following dependencies are required as part of RunPod platform functionality
 
 ### runpod.yaml
 
-Each container foulder needs to have a runpod.yaml file. This file will contain version info as well as services to be ran. The runpod.yaml file should be formatted as follows:
+Each container folder needs to have a runpod.yaml file. This file will contain version info as well as services to be ran. The runpod.yaml file should be formatted as follows:
 
 ```yaml
 version: '1.0.0'

official-templates/base/Dockerfile

@@ -208,3 +208,4 @@ RUN echo 'cat /etc/runpod.txt' >> /root/.bashrc
 RUN echo 'echo -e "\nFor detailed documentation and guides, please visit:\n\033[1;34mhttps://docs.runpod.io/\033[0m and \033[1;34mhttps://blog.runpod.io/\033[0m\n\n"' >> /root/.bashrc
 
 CMD ["/start.sh"]
+

official-templates/pytorch/Dockerfile

@@ -1,4 +1,4 @@
-ARG BASE_IMAGE
+ARG BASE_IMAGE=non-existing
 FROM ${BASE_IMAGE}
 
 ARG TORCH

official-templates/pytorch/README.md

@@ -7,7 +7,7 @@
 Example:
 
 ```bash
-cd containers/official-templates/pytorch
+cd official-templates/pytorch
 
 docker buildx bake 280-py311-cuda1281-cudnn-devel-ubuntu2204 --set 280-py311-cuda1281-cudnn-devel-ubuntu2204.platform=linux/amd64
 ```

official-templates/pytorch/docker-bake.hcl

@@ -19,6 +19,7 @@ group "default" {
         # CUDA 12.4.1
         "240-py311-cuda1241-devel-ubuntu2204",
         "251-py311-cuda1241-devel-ubuntu2204",
+        "260-py311-cuda1241-devel-ubuntu2204",
         # CUDA 12.8.1
         "280-py311-cuda1281-cudnn-devel-ubuntu2204",
 
@@ -63,6 +64,8 @@ group "cuda" {
         "220-py310-cuda1211-devel-ubuntu2204",
         "221-py310-cuda1211-devel-ubuntu2204",
         "240-py311-cuda1241-devel-ubuntu2204",
+        "251-py311-cuda1241-devel-ubuntu2204",
+        "260-py311-cuda1241-devel-ubuntu2204",
         "280-py311-cuda1281-cudnn-devel-ubuntu2204",
     ]
 }
@@ -158,7 +161,7 @@ target "220-py310-cuda1211-devel-ubuntu2204" {
     args = {
         BASE_IMAGE = "nvidia/cuda:12.1.1-devel-ubuntu22.04"
         PYTHON_VERSION = "3.10"
-        TORCH = "pip install torch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0"
+        TORCH = "torch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0"
     }
 }
 
@@ -173,7 +176,7 @@ target "221-py310-cuda1211-devel-ubuntu2204" {
     args = {
         BASE_IMAGE = "nvidia/cuda:12.1.1-devel-ubuntu22.04"
         PYTHON_VERSION = "3.10"
-        TORCH = "torch torchvision torchaudio"
+        TORCH = "torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 --index-url https://download.pytorch.org/whl/cu121"
     }
 }
 
@@ -188,7 +191,7 @@ target "240-py311-cuda1241-devel-ubuntu2204" {
     args = {
         BASE_IMAGE = "nvidia/cuda:12.4.1-devel-ubuntu22.04"
         PYTHON_VERSION = "3.11"
-        TORCH = "torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124"
+        TORCH = "torch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 --index-url https://download.pytorch.org/whl/cu124"
     }
 }
 
@@ -204,7 +207,22 @@ target "251-py311-cuda1241-devel-ubuntu2204" {
     args = {
         BASE_IMAGE = "nvidia/cuda:12.4.1-devel-ubuntu22.04"
         PYTHON_VERSION = "3.11"
-        TORCH = "torch torchvision torchaudio"
+        TORCH = "torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1"
+    }
+}
+
+target "260-py311-cuda1241-devel-ubuntu2204" {
+    dockerfile = "Dockerfile"
+    tags = ["${PUBLISHER}/pytorch:2.6.0-py3.11-cuda12.4.1-devel-ubuntu22.04"]
+    contexts = {
+        scripts = "../../container-template"
+        proxy = "../../container-template/proxy"
+        logo = "../../container-template"
+    }
+    args = {
+        BASE_IMAGE = "nvidia/cuda:12.4.1-devel-ubuntu22.04"
+        PYTHON_VERSION = "3.11"
+        TORCH = "torch==2.6.0 torchvision==0.21.0 torchaudio==2.6.0"
     }
 }
 
@@ -219,7 +237,7 @@ target "280-py311-cuda1281-cudnn-devel-ubuntu2204" {
     args = {
         BASE_IMAGE = "nvidia/cuda:12.8.1-cudnn-devel-ubuntu22.04"
         PYTHON_VERSION = "3.11"
-        TORCH = "torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128"
+        TORCH = "--pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128"
     }
 }