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

agunapal · web-flow · commit 784540311267 · 2022-12-12T15:26:53.000-08:00
* updated docker to build with CUDA 11.7 as default

* temp workflow for building and pushing docker images

* reverting tmp workflow
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(

Original file line number	Diff line number	Diff line change
`@@ -30,7 +30,7 @@`
`30`	`30`	`f"./build_image.sh -bt dev -t {organization}/torchserve:latest", dry_run`
`31`	`31`	`)`
`32`	`32`	`try_and_handle(`
`33`		`- f"./build_image.sh -bt dev -g -cv cu102 -t {organization}/torchserve:latest-gpu",`
	`33`	`+ f"./build_image.sh -bt dev -g -cv cu117 -t {organization}/torchserve:latest-gpu",`
`34`	`34`	`dry_run,`
`35`	`35`	`)`
`36`	`36`	`try_and_handle(`