Skip to content

Commit 32b6df5

Browse files
authored
chore: cover Nvidia T4/L4 GPU (#284)
1 parent 96ed3cd commit 32b6df5

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

sagemaker-entrypoint-cuda-all.sh

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,12 @@ get_compute_cap() {
5858
# Check if the GPU name contains "H100"
5959
elif [[ "$gpu_name" == *"H100"* ]]; then
6060
echo "90"
61+
# Cover Nvidia T4
62+
elif [[ "$gpu_name" == *"T4"* ]]; then
63+
echo "75"
64+
# Cover Nvidia L4
65+
elif [[ "$gpu_name" == *"L4"* ]]; then
66+
echo "89"
6167
else
6268
echo "80" # Default compute capability
6369
fi

0 commit comments

Comments
 (0)