back to torch2.5

dptech-corp · Jan 6, 2025 · dcb01bb · dcb01bb
1 parent cc24c8f
commit dcb01bb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/docker_rdma_latest.yml b/.github/workflows/docker_rdma_latest.yml
@@ -61,5 +61,5 @@ jobs:
           context: ./docker/rdma/
           push: true
           tags: |
-            dptechnology/unicore:latest-pytorch2.6.0-cuda12.6-rdma
-            dp-ve-registry-cn-beijing.cr.volces.com/dplc/unicore:latest-pytorch2.6.0-cuda12.6-rdma
+            dptechnology/unicore:latest-pytorch2.5.0-cuda12.5-rdma
+            dp-ve-registry-cn-beijing.cr.volces.com/dplc/unicore:latest-pytorch2.5.0-cuda12.5-rdma
diff --git a/docker/rdma/Dockerfile b/docker/rdma/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvcr.io/nvidia/pytorch:24.11-py3
+FROM nvcr.io/nvidia/pytorch:24.10-py3
 
 RUN APT_INSTALL="apt-get install -y --no-install-recommends" && \
     rm -rf /var/lib/apt/lists/* \