Use open drivers and modules-extra metapackages on ubuntu accelerators (

#2374)
GoogleCloudPlatform · Oct 4, 2024 · 2344911 · 2344911
1 parent be18a70
commit 2344911
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/daisy_workflows/image_build/accelerator_images/build_ubuntu_2204_accelerated.sh b/daisy_workflows/image_build/accelerator_images/build_ubuntu_2204_accelerated.sh
@@ -4,9 +4,9 @@ export DEBIAN_FRONTEND=noninteractive
 apt update -y || echo "BuildFailure"
 apt upgrade -y || echo "BuildFailure"
 # DKMS - not suitable for prod
-apt -y install nvidia-utils-550-server linux-modules-nvidia-550-server-gcp rdma-core || echo "BuildFailure"
+apt -y install nvidia-utils-550 linux-modules-nvidia-550-open-gcp rdma-core || echo "BuildFailure"
 # This is only for mlx5_ib - once that's in the main package stop including it
-apt -y install linux-modules-extra-$(uname -r) || echo "BuildFailure"
+apt -y install linux-modules-extra-gcp || echo "BuildFailure"
 tee /usr/bin/add-nvidia-repositories << EOF
 #!/bin/bash
 set -e

diff --git a/daisy_workflows/image_build/accelerator_images/build_ubuntu_2404_accelerated.sh b/daisy_workflows/image_build/accelerator_images/build_ubuntu_2404_accelerated.sh
@@ -4,9 +4,9 @@ export DEBIAN_FRONTEND=noninteractive
 apt update -y || echo "BuildFailure"
 apt upgrade -y || echo "BuildFailure"
 # DKMS - not suitable for prod
-apt -y install nvidia-utils-550-server linux-modules-nvidia-550-server-gcp rdma-core || echo "BuildFailure"
+apt -y install nvidia-utils-550 linux-modules-nvidia-550-open-gcp rdma-core || echo "BuildFailure"
 # This is only for mlx5_ib - once that's in the main package stop including it
-apt -y install linux-modules-extra-$(uname -r) || echo "BuildFailure"
+apt -y install linux-modules-extra-cp || echo "BuildFailure"
 tee /usr/bin/add-nvidia-repositories << EOF
 #!/bin/bash
 set -e