From 1a9c63ea3cccfef586080f6312f9685958dca285 Mon Sep 17 00:00:00 2001 From: Joe Date: Fri, 7 Jul 2023 13:54:14 +0900 Subject: [PATCH] Run GPU test in various environment --- .pfnci/config.pbtxt | 2 +- .pfnci/run_gpu_test.sh | 1 + .pfnci/run_gpu_test_all.sh | 10 ++++++++++ 3 files changed, 12 insertions(+), 1 deletion(-) create mode 100644 .pfnci/run_gpu_test_all.sh diff --git a/.pfnci/config.pbtxt b/.pfnci/config.pbtxt index 0d30a17..23f0ec6 100644 --- a/.pfnci/config.pbtxt +++ b/.pfnci/config.pbtxt @@ -14,6 +14,6 @@ configs { time_limit: { seconds: 3600 } - command: "bash .pfnci/run_gpu_test.sh" + command: "bash .pfnci/run_gpu_test_all.sh" } } diff --git a/.pfnci/run_gpu_test.sh b/.pfnci/run_gpu_test.sh index 6f1b1f6..31b8f14 100644 --- a/.pfnci/run_gpu_test.sh +++ b/.pfnci/run_gpu_test.sh @@ -11,6 +11,7 @@ set -eux nvidia-smi +export DEBIAN_FRONTEND=noninteractive apt-get update && apt-get install -y \ python3 \ python3-pip \ diff --git a/.pfnci/run_gpu_test_all.sh b/.pfnci/run_gpu_test_all.sh new file mode 100644 index 0000000..8156c17 --- /dev/null +++ b/.pfnci/run_gpu_test_all.sh @@ -0,0 +1,10 @@ +#!/bin/bash +set -eux + +for LIBTORCH_VERSION in 1.12.1 1.13.0; do + for OS in ubuntu20.04 ubuntu22.04; do + LIBTORCH_VERSION=$LIBTORCH_VERSION \ + OS=$OS \ + bash .pfnci/run_gpu_test.sh + done +done