mirror of
https://github.com/facebookresearch/faiss.git
synced 2025-06-03 21:54:02 +08:00
pin the dependecies version for x86_64 (#4046)
Summary: Pin dependecy version to get stable CI signal Pull Request resolved: https://github.com/facebookresearch/faiss/pull/4046 Reviewed By: mnorris11 Differential Revision: D66684832 Pulled By: junjieqi fbshipit-source-id: 9749d328688034514d6f2315d313fbf8045405ee
This commit is contained in:
parent
697b6ddf55
commit
cc5ff1ff0d
9
.github/actions/build_cmake/action.yml
vendored
9
.github/actions/build_cmake/action.yml
vendored
@ -32,7 +32,7 @@ runs:
|
||||
conda update -y -q conda
|
||||
echo "$CONDA/bin" >> $GITHUB_PATH
|
||||
|
||||
conda install -y -q python=3.11 cmake make swig "numpy<2" scipy pytest gflags
|
||||
conda install -y -q python=3.11 cmake=3.26 make=4.2 swig=4.0 "numpy<2" scipy=1.14 pytest=7.4 gflags=2.2
|
||||
|
||||
# install base packages for ARM64
|
||||
if [ "${{ runner.arch }}" = "ARM64" ]; then
|
||||
@ -42,7 +42,7 @@ runs:
|
||||
# install base packages for X86_64
|
||||
if [ "${{ runner.arch }}" = "X64" ]; then
|
||||
# TODO: merge this with ARM64
|
||||
conda install -y -q -c conda-forge gxx_linux-64 sysroot_linux-64
|
||||
conda install -y -q -c conda-forge gxx_linux-64=14.2 sysroot_linux-64=2.17
|
||||
conda install -y -q mkl=2023 mkl-devel=2023
|
||||
fi
|
||||
|
||||
@ -54,18 +54,17 @@ runs:
|
||||
conda install -y -q cuda-toolkit -c "nvidia/label/cuda-12.4.0"
|
||||
# and CUDA from cuVS channel for cuVS builds
|
||||
elif [ "${{ inputs.cuvs }}" = "ON" ]; then
|
||||
conda install -y -q libcuvs=24.08 cuda-version=12.4 cuda-toolkit -c rapidsai -c conda-forge -c "nvidia/label/cuda-12.4.0"
|
||||
conda install -y -q libcuvs=24.08 cuda-version=12.4 cuda-toolkit gxx_linux-64=12.4 -c rapidsai -c conda-forge -c "nvidia/label/cuda-12.4.0"
|
||||
fi
|
||||
|
||||
# install test packages
|
||||
conda install -y pytest
|
||||
if [ "${{ inputs.rocm }}" = "ON" ]; then
|
||||
: # 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<2.5" pytorch-cuda=12.4 -c pytorch -c nvidia/label/cuda-12.4.0
|
||||
else
|
||||
conda install -y -q pytorch -c pytorch
|
||||
conda install -y -q "pytorch<2.5" -c pytorch
|
||||
fi
|
||||
- name: ROCm - Install dependencies
|
||||
if: inputs.rocm == 'ON'
|
||||
|
Loading…
x
Reference in New Issue
Block a user