Skip to content

Latest commit

 

History

History
29 lines (21 loc) · 1.22 KB

README.md

File metadata and controls

29 lines (21 loc) · 1.22 KB
Raw

Implementation for the paper "Equivariant Flow Matching with Hybrid Probability Transport for 3D Molecule Generation"

Prerequisite

You will need to have a host machine with gpu, and have a docker with nvidia-container-runtime enabled.

Tip

  • This repo provide an easy to use script to install docker and nvidia-container-runtime, in ./MolFM/docker run sudo ./setup_docker_for_host.sh to setup your host machine.
  • You can also refer to install guide if you don't have them installed.

Quick start

Environment setup

Clone the repo with git clone,

git clone https://github.com/AlgoMole/MolFM.git

setup environment with docker,

cd ./MolFM/docker

make # a make is all you need

Note

  • The make will automatically build the docker image and run the container. with your host home directory mounted to the ${HOME}/home directory inside the container. highly recommended

  • If you need to setup the environment manually, please refer to files docker/Dockerfile, docker/asset/requirements.txt and docker/asset/apt_packages.txt.