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)
4 Sep 08, 2022
13 Nov 16, 2022
31 Dec 09, 2022
37 Jan 04, 2023
293 Dec 28, 2022
2 Dec 27, 2021
202 Dec 27, 2022
331 Jan 04, 2023
15 Dec 29, 2022
227 Jan 04, 2023
27 Dec 16, 2022
114 Nov 30, 2022
14 Dec 20, 2022
40 Sep 14, 2022
65 Oct 04, 2022
2.4k Dec 26, 2022
1 Dec 10, 2021
60 Oct 21, 2022
9 Nov 28, 2022