[python3-facenet-pytorch-pip] cpu version of torch in CI

tue-robotics · Jul 2, 2024 · 3e6b4cc · 3e6b4cc
1 parent 558c5b9
commit 3e6b4cc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/python3-facenet-pytorch-pip/install.bash b/python3-facenet-pytorch-pip/install.bash
@@ -0,0 +1,9 @@
+#! /usr/bin/env bash
+
+args=()
+if [[ "${CI}" == true ]]
+then
+    args+=("--extra-index-url" "https://download.pytorch.org/whl/cpu")
+fi
+
+tue-install-pip3 facenet-pytorch "${args{@]}"
diff --git a/python3-facenet-pytorch-pip/install.yaml b/python3-facenet-pytorch-pip/install.yaml