update files.

Author: freemankevin

.github/workflows/build.yml

@@ -26,4 +26,12 @@ jobs:
           context: .
           platforms: linux/amd64,linux/arm64
           push: true
-          tags: freelabspace/python-local:3.10
+          tags: |
+            freelabspace/python-local:3.10
+            freelabspace/python-local:3.10-torch${{ matrix.torch-version }}
+          build-args: |
+            TORCH_VERSION=${{ matrix.torch-version }}
+          # 定义不同的 torch 版本以便测试
+          matrix:
+            torch-version:
+              - "2.6.0"

Dockerfile

@@ -2,11 +2,6 @@
 ARG PYTHON_VERSION=3.10
 FROM python:${PYTHON_VERSION}-slim
 
-# 获取架构信息
-ARG TARGETPLATFORM
-ARG BUILDPLATFORM
-RUN echo "Building on $BUILDPLATFORM, targeting $TARGETPLATFORM"
-
 # 设置非交互式环境和上海时区
 ENV DEBIAN_FRONTEND=noninteractive
 RUN apt-get update -qq && \
@@ -29,6 +24,11 @@ RUN apt-get update -qq && \
 # 升级 pip，使用默认 PyPI 源
 RUN pip install -qq --upgrade pip
 
+# 定义 torch 版本参数
+ARG TORCH_VERSION
+# 安装 torch
+RUN pip install -qq --no-cache-dir torch==${TORCH_VERSION} --index-url https://download.pytorch.org/whl/torch
+
 # 设置工作目录
 WORKDIR /home