Merge pull request #171 from aws-deepracer-community/bugfix-revertgpu

revert gpu from 550 to 535
aws-deepracer-community · May 9, 2024 · 87af033 · 87af033
2 parents 573db00 + f10049a
commit 87af033
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-550 --no-install-recommends -o Dpkg::Options::="--force-overwrite"
+        sudo apt install -y nvidia-driver-535-server --no-install-recommends -o Dpkg::Options::="--force-overwrite"
         ;;
     *)
         echo "Unsupported distribution: $distribution"