mirrors
/
faiss
mirror of https://github.com/facebookresearch/faiss.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

bump pytorch to >=2

numpy2_hack
Amir Sadoughi 2024-08-26 13:39:43 -04:00
parent 8661c2381b
commit 82088e77a3
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
.github/actions/build_cmake/action.yml vendored
View File

@ -63,9 +63,9 @@ 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 pytorch-cuda=12.4 -c pytorch>=2 -c nvidia/label/cuda-12.4.0
else
conda install -y -q pytorch -c pytorch
conda install -y -q pytorch -c pytorch>=2
fi
- name: ROCm - Install dependencies
if: inputs.rocm == 'ON'