From 945f1b555e29ebc4b5326f23cf6d03b8937d0863 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Mon, 4 Mar 2024 04:09:26 -0500 Subject: [PATCH] set `NUM_WORKERS` to 0 in test_cuda action (#3404) It seems causing an OOM issue in the CUDA runner. Signed-off-by: Jinzhe Zeng --- .github/workflows/test_cuda.yml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/workflows/test_cuda.yml b/.github/workflows/test_cuda.yml index 915d983663..1a66bf5b82 100644 --- a/.github/workflows/test_cuda.yml +++ b/.github/workflows/test_cuda.yml @@ -51,10 +51,11 @@ jobs: - run: python -m pip install -v -e .[gpu,test,lmp,cu12,torch] "ase @ https://gitlab.com/ase/ase/-/archive/8c5aa5fd6448c5cfb517a014dccf2b214a9dfa8f/ase-8c5aa5fd6448c5cfb517a014dccf2b214a9dfa8f.tar.gz" env: DP_VARIANT: cuda - NUM_WORKERS: 0 DP_ENABLE_NATIVE_OPTIMIZATION: 1 - run: dp --version - run: python -m pytest source/tests --durations=0 + env: + NUM_WORKERS: 0 - name: Download libtorch run: | wget https://download.pytorch.org/libtorch/cu121/libtorch-cxx11-abi-shared-with-deps-2.2.1%2Bcu121.zip -O libtorch.zip