Skip to content

Commit 8346f37

Browse files
committed
Add cu124 for TRT_DOWNLOAD_LINK
1 parent f62541f commit 8346f37

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

packaging/pre_build_script_windows.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ choco install bazelisk -y
1010

1111
if [ ${CU_VERSION} = cu118 ]; then
1212
TRT_DOWNLOAD_LINK=https://developer.download.nvidia.com/compute/machine-learning/tensorrt/10.0.1/zip/TensorRT-10.0.1.6.Windows10.win10.cuda-11.8.zip
13-
elif [ ${CU_VERSION} = cu121 ]; then
13+
elif [ ${CU_VERSION} = cu121 ] || [ ${CU_VERSION} = cu124 ]; then
1414
TRT_DOWNLOAD_LINK=https://developer.download.nvidia.com/compute/machine-learning/tensorrt/10.0.1/zip/TensorRT-10.0.1.6.Windows10.win10.cuda-12.4.zip
1515
else
1616
echo "Unsupported CU_VERSION"

0 commit comments

Comments
 (0)