From ebe5a692f89c08b835c544a25631d29fdea0ff63 Mon Sep 17 00:00:00 2001 From: Amir Sadoughi Date: Tue, 22 Oct 2024 13:35:02 -0700 Subject: [PATCH] Update RAFT CI with pytorch 2.4.1 (#3980) Summary: Related to testing in https://github.com/facebookresearch/faiss/pull/3974 Based on comparing the logs of two runs: - failing: https://github.com/facebookresearch/faiss/actions/runs/11409771344/job/31751246207 - passing: https://github.com/facebookresearch/faiss/actions/runs/11368781432/job/31625550227 Pull Request resolved: https://github.com/facebookresearch/faiss/pull/3980 Reviewed By: junjieqi Differential Revision: D64778154 Pulled By: asadoughi fbshipit-source-id: f4e53fed3850f3e0f391015c0349ee14da68330a --- .github/actions/build_cmake/action.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/actions/build_cmake/action.yml b/.github/actions/build_cmake/action.yml index 2e32dc1578..1ec1186484 100644 --- a/.github/actions/build_cmake/action.yml +++ b/.github/actions/build_cmake/action.yml @@ -63,7 +63,7 @@ runs: : # skip torch install via conda, we need to install via pip to get # ROCm-enabled version until it's supported in conda by PyTorch elif [ "${{ inputs.gpu }}" = "ON" ]; then - conda install -y -q pytorch pytorch-cuda=12.4 -c pytorch -c nvidia/label/cuda-12.4.0 + conda install -y -q "pytorch<2.5" pytorch-cuda=12.4 -c pytorch -c nvidia/label/cuda-12.4.0 else conda install -y -q pytorch -c pytorch fi