Running AlphaFold2 (from ColabFold) in Azure Machine Learning
Colby T. Ford, Ph.D.
Companion repository for Medium Post: How to predict many protein structures with AlphaFold2 at-scale in Azure Machine Learning.
Resources:
- Jupyter Notebook: AzureML_AlphaFold2_HyperDrive.ipynb
- Sample Sequence FASTA: sequences.fasta
- Dockerfile: Dockerfile
Docker Image Creation
The custom Docker image for running ColabFold in Azure ML HyperDrive is available on Docker Hub at https://hub.docker.com/r/cford38/alphafold2_aml. If you'd like to build the image for yourself, follow the instructions below.
First, build the custom Azure ML Docker image (with the required dependencies for the AlphaFold2 system).
docker build -t alphafold2_aml --build-arg IMAGE_VERSION=$(git rev-parse --short HEAD) .
# docker run --name alphafold2_aml --rm -p 8787:8787 alphafold2_aml
# docker exec -it alphafold2_aml /bin/bash
To push the image to DockerHub, run the following:
docker image tag alphafold2_aml <USERNAME>/alphafold2_aml:latest
docker push <USERNAME>/alphafold2_aml:latest
# docker image tag alphafold2_aml cford38/alphafold2_aml:latest
# docker push cford38/alphafold2_aml:latest
To push the image to an Azure Container Registry, run the following:
az login
az account set --subscription <SUBSCRIPTION_ID>
az acr login --name <CONTAINER_REGISTRY_NAME>
docker tag mmae_aml <CONTAINER_REGISTRY_NAME>.azurecr.io/alphafold2_aml:$(git rev-parse --short HEAD)
docker push <CONTAINER_REGISTRY_NAME>.azurecr.io/alphafold2_aml:$(git rev-parse --short HEAD)
2 Dec 12, 2021
688 Jan 04, 2023
19 Dec 10, 2021
3.9k Jan 04, 2023
709 Dec 31, 2022
1.5k Jan 01, 2023
2 Jan 17, 2022
35 Dec 14, 2022
441 Dec 05, 2022
22 Dec 11, 2022
40 Oct 30, 2022
421 Jan 04, 2023
42 Jul 25, 2022
1 Apr 25, 2022
94 Nov 12, 2022
873 Jan 04, 2023
21 Dec 29, 2022
527 Dec 04, 2022
80 Dec 14, 2022
8 Dec 12, 2022