error in geneval

[quantumiracle]

Hi,

I'm running the eval code for Geneval env, and got error:

mmcv/ops/multi_scale_deform_attn.py", line 319, in forward
    assert (spatial_shapes[:, 0] * spatial_shapes[:, 1]).sum() == num_value
AssertionError

Is this caused by wrong dependency, e.g. mmcv? I'm following the exact instruction for geneval env setup.

Thanks

[Franklin-Zhang0]

I'm working fine with this installation scirpt on 8xA100 machine. Maybe you can refer to this issue, where they talked about more detailed installation guides and some issues on h100 machines.

[quantumiracle]

Solved. Needs Cuda compiling for mmcv-full.

pip install torch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu121

export CUDA_HOME=/usr/local/cuda-12.8
export PATH=$CUDA_HOME/bin:$PATH
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH

git clone https://github.com/open-mmlab/mmcv.git
cd mmcv; git checkout 1.x
MMCV_WITH_OPS=1 MMCV_CUDA_ARGS="-arch=sm_90" pip install -v -e .