Add pytorch-gpu for security patching (#1619)

deepjavalibrary · Mar 11, 2024 · 87d3190 · 87d3190
1 parent d3a2e13
commit 87d3190
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/serving/docker/scripts/security_patch.sh b/serving/docker/scripts/security_patch.sh
@@ -4,7 +4,8 @@ IMAGE_NAME=$1
 
 apt-get update
 
-if [[ "$IMAGE_NAME" == "deepspeed" ]]; then
+if [[ "$IMAGE_NAME" == "deepspeed" ]] || \
+   [[ "$IMAGE_NAME" == "pytorch-gpu" ]]; then
   apt-get upgrade -y dpkg openssl curl libssl3
 elif [[ "$IMAGE_NAME" == "cpu" ]]; then
   apt-get upgrade -y libpcre2-8-0 libdbus-1-3 curl