This Docker image includes the per-trained DiffDock and ESM models and it takes 18.54gb of storage after downloading. Follow the instructions below to run it using either CPU or CUDA-enabled GPU. The Docker image is of DiffDock-L (v1.1) that was built on 04.15.24 with the latest updates through GitHub.
docker pull externelly/diffdock:latest
# To run DiffDock with CPU only
docker run --name=diffdock -v ${PWD}/mount:/DiffDock/mount --rm -it externelly/diffdock:latest
# To run DiffDock with CUDA GPU enabled
docker run --gpus all --name=diffdock -v ${PWD}/mount:/DiffDock/mount --rm -it externelly/diffdock:latest
Create a pdbs/ folder in your mount folder and copy all PDB files into it.
Prepare a mount/diffdock_analysis.csv CSV file in the mount folder with the following columns:
protein_path,ligand_description,complex_name,protein_sequence
mount/pdbs/PDB-id.pdb,SMLES,PDB-id_SMILES-id,
...
# one row for each PDB-SMILES combination
Run the followin command to download models: python -m inference.
It will takes a couple of minutes. Not persistant once you close the docker container.
python -m inference --config default_inference_args.yaml --protein_ligand_csv mount/diffdock_analysis.csv --out_dir mount/results
Don't forget to cite the original paper.
DiffDock-L
Corso, G., Deng, A., Fry, B., Polizzi, N., Barzilay, R., & Jaakkola, T. (2024). Deep Confident Steps to New Pockets: Strategies for Docking Generalization. arXiv preprint arXiv:2402.18396.
DiffDock
Corso, G., Stärk, H., Jing, B., Barzilay, R., & Jaakkola, T. (2022). Diffdock: Diffusion steps, twists, and turns for molecular docking. arXiv preprint arXiv:2210.01776.
Content type
Image
Digest
sha256:ab9484806…
Size
10.1 GB
Last updated
over 2 years ago
docker pull externelly/diffdock