Fix docker build script to build with CUDA 11.7 (#2032)

* updated docker to build with CUDA 11.7 as default * temp workflow for building and pushing docker images * reverting tmp workflow
pytorch · Dec 12, 2022 · 7845403 · 7845403
1 parent ddaddb9
commit 7845403
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/build_upload_release.py b/docker/build_upload_release.py
@@ -30,7 +30,7 @@
         f"./build_image.sh -bt dev -t {organization}/torchserve:latest", dry_run
     )
     try_and_handle(
-        f"./build_image.sh -bt dev -g -cv cu102 -t {organization}/torchserve:latest-gpu",
+        f"./build_image.sh -bt dev -g -cv cu117 -t {organization}/torchserve:latest-gpu",
         dry_run,
     )
     try_and_handle(