Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Update MMSeqs image for GPU support #9

Open
keiran-rowell-unsw opened this issue Jan 21, 2025 · 0 comments
Open

Update MMSeqs image for GPU support #9

keiran-rowell-unsw opened this issue Jan 21, 2025 · 0 comments
Assignees
Labels
enhancement New feature or request

Comments

@keiran-rowell-unsw
Copy link
Member

keiran-rowell-unsw commented Jan 21, 2025

Description of feature

GPU execution of MSA construction using MMSeqs has a 170x speedup compared to AlphaFold2's default JackHMMER execution.

https://developer.nvidia.com/blog/boost-alphafold2-protein-structure-prediction-with-gpu-accelerated-mmseqs2/

I've put GPU formatted MMSeqs DBs on Katana and this bares out on our system. The gpu-server mode takes up about 50GB of VRAM so is suitable for H200 use but not A100. Nevertheless the MMSeqs-GPU speedup is still dramatic even without an index GPU server, provided there's fast local storage.

The MMSeqs team are releasing CUDA12 compatible docker images, for T4 -> HX00 GPUs. Try to apptainer pull from that rather than roll a separate .sif for proteinfold, though some setups may need compilation with CUDA11 compatibility, or particular CPU vector extensions, etc
https://github.com/soedinglab/MMseqs2/pkgs/container/mmseqs2

@keiran-rowell-unsw keiran-rowell-unsw added the enhancement New feature or request label Jan 21, 2025
@keiran-rowell-unsw keiran-rowell-unsw self-assigned this Jan 21, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

1 participant