show cudnn version

microsoft · Feb 22, 2024 · f00c993 · f00c993
1 parent 2b68e44
commit f00c993
Showing 1 changed file with 7 additions and 1 deletion.
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
@@ -223,8 +223,14 @@ stages:
             -e NVIDIA_TF32_OVERRIDE=0 \
             $(Repository) \
             /bin/bash -c '
-              set -ex; \
               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; \
               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) ; \