bowang-lab / BIONIC

Biological Network Integration using Convolutions
MIT License
58 stars 4 forks source link

Refactor #3

Closed duncster94 closed 4 years ago

duncster94 commented 4 years ago

This PR completely refactors BIONIC, resulting in the following improvements:

  1. Updates BIONIC to the newest versions of PyTorch and PyTorch Geometric.
  2. Adds testing.
  3. Adds CPU support.
  4. Improves code quality.
  5. Improves documentation and adds typing.
  6. Adds a build workflow using GitHub Actions.

Wheels are made available for CPU, CUDA 9.2, CUDA 10.1 and CUDA 10.2 versions of BIONIC on Linux and CPU, CUDA 10.1 and CUDA 10.2 versions of BIONIC on Windows.

Built wheels are (manually) uploaded to the Wang Lab datacenter here. In the future I will make this process automatic.

Wheels for a given CUDA distribution can be installed with $ pip install bionic_model==${VERSION}+${CUDA} -f https://bowang-lab.github.io/BIONIC/wheels.html where ${VERSION} is the version of BIONIC (currently 0.1.0) and ${CUDA} is the CUDA distribution (one of cpu, cu92, cu101 or cu102). Note that cu92 is not available for Windows.