google-deepmind / alphafold

Open source code for AlphaFold 2.
Apache License 2.0
12.91k stars 2.29k forks source link

optimize alphafold2 speed on different GPUs #950

Open gorliver opened 5 months ago

gorliver commented 5 months ago

Hi, I have run alphafold2 on two machines with A800 and V100, both machines has NVME ssd. Alphafold2 on A800 is 3-5 times slower than V100, which is not expected as A800 is a newer GPU with higher computational power. I wonder there are any configurations of alphafold2 to maximize the use of different GPUs?