google-deepmind / google-deepmind/alphafold

Does CUDA version matter?

Open
#710 3 comments 0 reactions 0 assignees View on GitHub
setup
Dominant language
Python
Stars
14.9k
Forks
2.9k
PR merge metrics
No merged PRs in 30d

Description

I checked if Alphafold can use the GPU by running the following:
docker run --rm --gpus all nvidia/cuda:11.0-base nvidia-smi

I get the following error:
Unable to find image 'nvidia/cuda:11.0-base' locally
docker: Error response from daemon: manifest for nvidia/cuda:11.0-base not found: manifest unknown: manifest unknown.
See 'docker run --help'.

However, I know I have a newer version of cuda. If I run the following, everything looks great:
docker run --rm --gpus all nvidia/cuda:11.6.2-base-ubuntu20.04 nvidia-smi

Will alphafold only recognize a specific version of CUDA, or is this good enough to run alphafold?

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.