nividia-driver-550 for Ubuntu22.04 (#167)

* test recommended drivers with aws g6 * revert ubuntu2004 --------- Co-authored-by: Mark <[email protected]>
aws-deepracer-community · May 3, 2024 · e7c3505 · e7c3505
1 parent 12832da
commit e7c3505
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bin/prepare.sh b/bin/prepare.sh
@@ -58,7 +58,7 @@ if [[ "${ARCH}" == "gpu" && -z "${IS_WSL2}" ]]; then
         sudo apt install -y nvidia-driver-525-server --no-install-recommends -o Dpkg::Options::="--force-overwrite"
         ;;
     ubuntu2204)
-        sudo apt install -y nvidia-driver-535-server --no-install-recommends -o Dpkg::Options::="--force-overwrite"
+        sudo apt install -y nvidia-driver-550 --no-install-recommends -o Dpkg::Options::="--force-overwrite"
         ;;
     *)
         echo "Unsupported distribution: $distribution"