Force L0_perf_client to run on single GPU (#789)

triton-inference-server · Oct 24, 2019 · 7fcf629 · 7fcf629
1 parent 431b9ee
commit 7fcf629
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/qa/L0_perf_client/test.sh b/qa/L0_perf_client/test.sh
@@ -35,6 +35,8 @@ if [ -z "$REPO_VERSION" ]; then
     exit 1
 fi
 
+export CUDA_VISIBLE_DEVICES=0
+
 CLIENT_LOG="./perf_client.log"
 PERF_CLIENT=../clients/perf_client