We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f62541f commit 8346f37Copy full SHA for 8346f37
packaging/pre_build_script_windows.sh
@@ -10,7 +10,7 @@ choco install bazelisk -y
10
11
if [ ${CU_VERSION} = cu118 ]; then
12
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
+elif [ ${CU_VERSION} = cu121 ] || [ ${CU_VERSION} = cu124 ]; then
14
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
15
else
16
echo "Unsupported CU_VERSION"
0 commit comments