From 1afb81fc949f02f9db0efc159436f4e37f0c1d7d Mon Sep 17 00:00:00 2001 From: smg3d Date: Sun, 24 Nov 2024 11:51:45 +0000 Subject: [PATCH] Update GPU compute capability check to reflect working versions - Add explicit check for compute capability < 6.0 - Keep check for range [7.0, 8.0) - Update error message to clarify working versions (6.x and 8.x) - Addresses issue #59 --- run_alphafold.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/run_alphafold.py b/run_alphafold.py index 0c0230a..1f44ee5 100644 --- a/run_alphafold.py +++ b/run_alphafold.py @@ -613,10 +613,16 @@ def main(_): if _RUN_INFERENCE.value: # Fail early on incompatible devices, but only if we're running inference. gpu_devices = jax.local_devices(backend='gpu') - if gpu_devices and float(gpu_devices[0].compute_capability) < 8.0: + compute_capability = float(gpu_devices[0].compute_capability) + if gpu_devices and ( + compute_capability < 6.0 + or (compute_capability >= 7.0 and compute_capability < 8.0) + ): raise ValueError( 'There are currently known unresolved numerical issues with using' - ' devices with compute capability less than 8.0. See ' + ' devices with compute capability 7.x. The code has not been tested' + ' with compute capability < 6.0. The code has been confirmed to work' + ' with compute capability 6.x and 8.x. See ' ' https://github.com/google-deepmind/alphafold3/issues/59 for' ' tracking.' )