"Prediction failed" when createdb --gpu 1

[sevengo8378]

I want to create a db based on Prostt5, and first use SwissProt to test it, and I encounter the error "Prediction failed". The execution command is as follows: ./foldseek createdb ./uniprot_sprot.fasta sp/db --prostt5-model foldseek_db/weights --gpu 1

Expected Behavior

The database is successfully generated in the sp/directory

Current Behavior

Steps to Reproduce (for bugs)

conda create -n foldseek-gpu -c conda-forge cmake cuda-nvcc libcurand-dev libcublas-dev cuda-nvrtc-dev cuda-version=12.4 conda activate foldseek-gpu cmake -DCMAKE_BUILD_TYPE=Debug -DENABLE_CUDA=1 -DCUDAToolkit_ROOT=$(dirname $(which nvcc))/../targets/x86_64-linux .. cd $work_dir ln -sf ./foldseek_github/build/src/foldseek ./foldseek ./foldseek createdb ./uniprot_sprot.fasta sp/db --prostt5-model foldseek_db/weights --gpu 1

Additional info 1: According to https://github.com/steineggerlab/foldseek/issues/285, I also compiled the debug version and debugged it as follows

gdb --args ./foldseek_debug createdb ./uniprot_sprot.fasta sp/db --prostt5-model foldseek_db/weights --gpu 1 
# wait for a prompt to appear
r
# wait for the crash
bt

Here is the result: error_log.txt

Additional info 2: If I pick the first 10,000 sequences of Swissprot as sp_1w.fasta as input, createdb is finally successful

Your Environment

• Operating system and version: Linux xinmatrix 6.5.0-27-generic #28~22.04.1-Ubuntu SMP PREEMPT_DYNAMIC Fri Mar 15 10:51:06 UTC 2 x86_64 x86_64 x86_64 GNU/Linux