Update tools/ci_build/github/azure-pipelines/linux-gpu-ci-pipeline.yml

Co-authored-by: Tianlei Wu <[email protected]>
microsoft · Feb 22, 2024 · 5a0877e · 5a0877e
1 parent 53c72d4
commit 5a0877e
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/tools/ci_build/github/azure-pipelines/linux-gpu-ci-pipeline.yml b/tools/ci_build/github/azure-pipelines/linux-gpu-ci-pipeline.yml
@@ -224,13 +224,9 @@ stages:
             $(Repository) \
             /bin/bash -c '
               nvidia-smi; \
-              /sbin/ldconfig -N -v $(sed "s/:/ /" <<< $LD_LIBRARY_PATH) 2>/dev/null | grep libcudart; \
-              /sbin/ldconfig -N -v $(sed "s/:/ /" <<< $LD_LIBRARY_PATH) 2>/dev/null | grep libcudnn; \
-              /sbin/ldconfig -N -v $(sed "s/:/ /" <<< $LD_LIBRARY_PATH) 2>/dev/null | grep libnvinfer; \
-              whereis cudnn; \
-              whereis cuda; \
-              cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2; \
-              cat /usr/include/x86_64-linux-gnu/cudnn_v*.h | grep CUDNN_MAJOR -A 2; \
+              /sbin/ldconfig -N -v $(sed "s/:/ /" <<< $LD_LIBRARY_PATH) 2>/dev/null | grep -E "libcudart.so|libcudnn.so|libnvinfer.so"; \
+              cat /usr/local/cuda/include/cuda.h | grep -m1 CUDA_VERSION; \
+              cat /usr/include/cudnn_version.h | grep CUDNN_MAJOR -m1 -A 2; \
               ln -s /opt/python/cp38-cp38/bin/python3 /tmp/python3; \
               /tmp/python3 -m pip install /build/Release/dist/*.whl; \
               /tmp/python3 -u -c "from onnxruntime.capi._pybind_state import (OrtDevice as C_OrtDevice) ; \