cache 4

pytorch · Sep 15, 2024 · 470b285 · 470b285
1 parent fead916
commit 470b285
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/docker/docker_nightly.py b/docker/docker_nightly.py
@@ -40,10 +40,10 @@
 
     # Build Nightly images and append the date in the name
     try_and_handle(f"./build_image.sh -m -n -t {organization}/{cpu_version}", dry_run)
-    try_and_handle(
-        f"./build_image.sh -g -cv cu121 -n -t {organization}/{gpu_version}",
-        dry_run,
-    )
+    # try_and_handle(
+    #     f"./build_image.sh -g -cv cu121 -n -t {organization}/{gpu_version}",
+    #     dry_run,
+    # )
     # try_and_handle(
     #     f"./build_image.sh -bt dev -cpp -t {organization}/{cpp_dev_cpu_version}",
     #     dry_run,
@@ -54,7 +54,7 @@
     # )
 
     # Push Nightly images to official PyTorch Dockerhub account
-    try_and_handle(f"docker push {organization}/{gpu_version}", dry_run)
+    # try_and_handle(f"docker push {organization}/{gpu_version}", dry_run)
     # try_and_handle(f"docker push {organization}/{cpp_dev_cpu_version}", dry_run)
     # try_and_handle(f"docker push {organization}/{cpp_dev_gpu_version}", dry_run)
 
@@ -63,10 +63,10 @@
         f"docker buildx imagetools create --tag {organization}/{project}:latest-cpu {organization}/{cpu_version}",
         dry_run,
     )
-    try_and_handle(
-        f"docker buildx imagetools create --tag {organization}/{project}:latest-gpu {organization}/{gpu_version}",
-        dry_run,
-    )
+    # try_and_handle(
+    #     f"docker buildx imagetools create --tag {organization}/{project}:latest-gpu {organization}/{gpu_version}",
+    #     dry_run,
+    # )
     # try_and_handle(
     #     f"docker tag {organization}/{cpp_dev_cpu_version} {organization}/{project}:latest-cpp-dev-cpu",
     #     dry_run,