-

mila-iqia · Jun 17, 2024 · 20c9ab3 · 20c9ab3
1 parent 5ba296f
commit 20c9ab3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -25,7 +25,7 @@ hrepr = "^0.4.0"
 blessed = "^1.19.1"
 pathspec = "^0.9.0"
 cp-template = "^0.3.0"
-pandas = "^1.4.2"
+pandas = ">=1.4.2"
 numpy = ">=1.23.0"
 pynvml = "^11.4.1"
 tqdm = "^4.64.1"

diff --git a/scripts/article/run_batch_x_worker.sh b/scripts/article/run_batch_x_worker.sh
@@ -24,7 +24,13 @@ if [ "$DRY" -eq 0 ]; then
     #
     source $MILABENCH_WORDIR/env/bin/activate
 fi
-
+export MILABENCH_GPU_ARCH=cuda
+export MILABENCH_WORDIR="$(pwd)/$MILABENCH_GPU_ARCH"
+export MILABENCH_BASE="$MILABENCH_WORDIR/results"
+export MILABENCH_CONFIG="$MILABENCH_WORDIR/milabench/config/standard.yaml"
+export MILABENCH_VENV="$MILABENCH_WORDIR/env"
+export BENCHMARK_VENV="$MILABENCH_WORDIR/results/venv/torch"
+export MILABENCH_PREPARE=0
 
 source $MILABENCH_WORDIR/env/bin/activate