Run GPU test in various environment

pfnet · Jul 7, 2023 · 1a9c63e · 1a9c63e
1 parent 6f0d686
commit 1a9c63e
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 1 deletion.
diff --git 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
@@ -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
@@ -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