The installation error is as follows: （follow the guide ，after pip install rankify[all]）

[weblwh]

Downloading https://mirrors.tencent.com/pypi/packages/11/34/9bf60e736ed7bbe15055ac2dab48ec67d9dbd088d2b4ae318fd77190ab4e/flash_attn-2.7.4.post1.tar.gz (6.0 MB)
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 6.0/6.0 MB 6.3 MB/s eta 0:00:00
Preparing metadata (setup.py) ... error
error: subprocess-exited-with-error

× python setup.py egg_info did not run successfully.
│ exit code: 1
╰─> [19 lines of output]
/tmp/pip-install-1yacv3jy/flash-attn_9aff4fc00be24c07984f207c1aad7ce4/setup.py:106: UserWarning: flash_attn was requested, but nvcc was not found. Are you sure your environment has nvcc available? If you're installing within a container from https://hub.docker.com/r/pytorch/pytorch, only images whose names contain 'devel' will provide nvcc.
warnings.warn(
Traceback (most recent call last):
File "", line 2, in
File "", line 34, in
File "/tmp/pip-install-1yacv3jy/flash-attn_9aff4fc00be24c07984f207c1aad7ce4/setup.py", line 198, in
CUDAExtension(
File "/root/miniconda3/envs/rankify/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1078, in CUDAExtension
library_dirs += library_paths(cuda=True)
File "/root/miniconda3/envs/rankify/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1209, in library_paths
if (not os.path.exists(_join_cuda_home(lib_dir)) and
File "/root/miniconda3/envs/rankify/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 2416, in _join_cuda_home
raise OSError('CUDA_HOME environment variable is not set. '
OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.

  torch.__version__  = 2.5.1+cu124


  [end of output]

note: This error originates from a subprocess, and is likely not a problem with pip.
error: metadata-generation-failed

× Encountered error while generating package metadata.
╰─> See above for output.

note: This is an issue with the package mentioned above, not pip.
hint: See above for details.

[abdoelsayed2016]

Hello

it's recommended to install the CUDA version 12.4 or 12.6 and then install

conda/conda#7757

facebookresearch/sam2#41

Then we installing Rankify with PyTorch 2.5.1. Refer to the PyTorch installation page for platform-specific installation commands.

If you have access to GPUs, it's recommended to install the CUDA version 12.4 or 12.6 of PyTorch, as many of the evaluation metrics are optimized for GPU use.

To install Pytorch 2.5.1 you can install it from the following cmd

pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu124

[abdoelsayed2016]

@weblwh

did you solve it?