Environment installation fails on A40 with CUDA 12.4 during ./build.sh libraft

[CraneQinghe]

Description

I tried to set up the environment following the provided installation instructions, but the build fails at the last step (./build.sh libraft).

Environment

• GPU: NVIDIA A40
• CUDA version: 12.4
• OS: Linux
• Python: 3.10 (conda environment)
• Conda: latest

Steps to Reproduce

# Setup conda environment
conda create -yn quest python=3.10
conda activate quest

# Install Quest
cd /data/haiqwa/zevin_nfs/code/qinghe/quest
pip install -e .

# Install Flash-Attention
pip install ninja packaging
cd /data/haiqwa/zevin_nfs/code/qinghe/quest/flash-attention
pip install . --no-build-isolation --use-pep517 --no-deps

# Install CMake
conda install cmake

# Build libraft
cd /data/haiqwa/zevin_nfs/code/qinghe/quest/kernels/3rdparty/raft
./build.sh libraft

The build fails with the following CMake errors:

Building for the architecture of the GPU in the system...
CMake Error at CMakeLists.txt:18 (include):
  include could not find requested file:

    rapids-cmake

CMake Error at CMakeLists.txt:19 (include):
  include could not find requested file:

    rapids-cpm

CMake Error at CMakeLists.txt:20 (include):
  include could not find requested file:

    rapids-export

CMake Error at CMakeLists.txt:21 (include):
  include could not find requested file:

    rapids-find

CMake Error at CMakeLists.txt:30 (include):
  include could not find requested file:

    rapids-cuda

CMake Error at CMakeLists.txt:31 (rapids_cuda_init_architectures):
  Unknown CMake command "rapids_cuda_init_architectures".

-- Configuring incomplete, errors occurred!

I tried setting environment variables to point to rapids-cmake, but it still fails:

export CMAKE_MODULE_PATH=$PWD/rapids-cmake/rapids-cmake/rapids-cmake:$CMAKE_MODULE_PATH
export CMAKE_PREFIX_PATH=$PWD/rapids-cmake/rapids-cmake/rapids-cmake:$CMAKE_PREFIX_PATH

test -f $PWD/rapids-cmake/rapids-cmake/rapids-cmake/rapids-cmake.cmake && echo "Found rapids-cmake"
# Output: Found rapids-cmake