From 791e95dba706d0f052c6426323614be01b9ca70d Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Mon, 15 Apr 2024 22:52:20 -0400 Subject: [PATCH] fix test_cuda.yml Signed-off-by: Jinzhe Zeng --- .github/workflows/test_cuda.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/test_cuda.yml b/.github/workflows/test_cuda.yml index 22fa5ccb5b..db6b6bbbaa 100644 --- a/.github/workflows/test_cuda.yml +++ b/.github/workflows/test_cuda.yml @@ -47,10 +47,10 @@ jobs: && sudo apt-get -y install cuda-12-3 libcudnn8=8.9.5.*-1+cuda12.3 if: false # skip as we use nvidia image - run: python -m pip install -U uv - - run: python -m uv pip install "tensorflow>=2.15.0rc0" "torch>=2.2.0" + - run: python -m uv pip install --system "tensorflow>=2.15.0rc0" "torch>=2.2.0" - run: | export TENSORFLOW_ROOT=$(python -c 'import importlib,pathlib;print(pathlib.Path(importlib.util.find_spec("tensorflow").origin).parent)') - python -m uv pip install -v -e .[gpu,test,lmp,cu12,torch] mpi4py "ase @ https://gitlab.com/ase/ase/-/archive/8c5aa5fd6448c5cfb517a014dccf2b214a9dfa8f/ase-8c5aa5fd6448c5cfb517a014dccf2b214a9dfa8f.tar.gz" + python -m uv pip install --system -v -e .[gpu,test,lmp,cu12,torch] mpi4py "ase @ https://gitlab.com/ase/ase/-/archive/8c5aa5fd6448c5cfb517a014dccf2b214a9dfa8f/ase-8c5aa5fd6448c5cfb517a014dccf2b214a9dfa8f.tar.gz" env: DP_VARIANT: cuda DP_ENABLE_NATIVE_OPTIMIZATION: 1