google-deepmind / google-deepmind/alphafold
Does CUDA version matter?
- 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
Assessment
This issue has not been assessed yet.