diff --git a/sagemaker-entrypoint-cuda-all.sh b/sagemaker-entrypoint-cuda-all.sh
index 4cf645ff..a3c63cbb 100644
--- a/sagemaker-entrypoint-cuda-all.sh
+++ b/sagemaker-entrypoint-cuda-all.sh
@@ -58,6 +58,12 @@ get_compute_cap() {
     # Check if the GPU name contains "H100"
     elif [[ "$gpu_name" == *"H100"* ]]; then
         echo "90"
+    # Cover Nvidia T4
+    elif [[ "$gpu_name" == *"T4"* ]]; then
+        echo "75"
+    # Cover Nvidia L4
+    elif [[ "$gpu_name" == *"L4"* ]]; then
+        echo "89"
     else
         echo "80"  # Default compute capability
     fi