From 08e5f89b37fef05343ab7e823259f4d4f571c097 Mon Sep 17 00:00:00 2001 From: Changming Sun Date: Mon, 1 Jun 2020 20:42:15 -0700 Subject: [PATCH] Fix the nuget gpu pipeline (#4106) --- .../runtest-docker-gpu.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/csharp/test/Microsoft.ML.OnnxRuntime.EndToEndTests/runtest-docker-gpu.sh b/csharp/test/Microsoft.ML.OnnxRuntime.EndToEndTests/runtest-docker-gpu.sh index 81dd88b9bf..562ca596af 100755 --- a/csharp/test/Microsoft.ML.OnnxRuntime.EndToEndTests/runtest-docker-gpu.sh +++ b/csharp/test/Microsoft.ML.OnnxRuntime.EndToEndTests/runtest-docker-gpu.sh @@ -35,6 +35,6 @@ sudo --preserve-env docker run -h $HOSTNAME \ -e "PackageName=$PackageName" \ -e "RunTestCsharp=$RunTestCsharp" \ -e "RunTestNative=$RunTestNative" \ - "onnxruntime-$IMAGE" \ + onnxruntime-ubuntu16.04-cuda10.1-cudnn7.6 \ /bin/bash /onnxruntime_src/csharp/test/Microsoft.ML.OnnxRuntime.EndToEndTests/runtest.sh \ /home/onnxruntimedev/$NUGET_REPO_DIRNAME /onnxruntime_src /home/onnxruntimedev $CurrentOnnxRuntimeVersion