From 00eb7166ad91bf7845309fe73c2ee1be8ebb4e01 Mon Sep 17 00:00:00 2001 From: yf711 Date: Wed, 22 Nov 2023 13:59:21 -0800 Subject: [PATCH] test --- .../linux/docker/Dockerfile.ubuntu_cuda11_8_tensorrt8_6 | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda11_8_tensorrt8_6 b/tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda11_8_tensorrt8_6 index f586b1f392b17..cb0069fce4515 100644 --- a/tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda11_8_tensorrt8_6 +++ b/tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda11_8_tensorrt8_6 @@ -41,6 +41,12 @@ RUN v="8.6.1.6-1+cuda11.8" &&\ # Compile trtexec RUN cd /usr/src/tensorrt/samples/trtexec && make +# Check GPU status +RUN nvidia-smi -i 0 -q -d CLOCK +RUN nvidia-smi -i 0 -pm 1 +RUN nvidia-smi -i 0 -q -d CLOCK | grep -A 4 "Max Clocks" | grep "Graphics" | awk '{print $3}' +RUN nvidia-smi --query-gpu=gpu_name,gpu_bus_id,vbios_version,clocks.current.sm,clocks.mem, --format=csv + # Install Valgrind RUN apt-get install -y valgrind @@ -66,7 +72,7 @@ ARG ONNXRUNTIME_BRANCH=main ARG ONNXRUNTIME_COMMIT_ID # The supported CUDA architecture -ARG CMAKE_CUDA_ARCHITECTURES=37;50;52;60;61;70;75;80 +ARG CMAKE_CUDA_ARCHITECTURES=75 WORKDIR ${ONNXRUNTIME_LOCAL_CODE_DIR}