In this file, I'll list out all of the packages that you need to install via conda. For each package, I'll give it in the form "X:x
", where X is the package name and x
is what you type in the conda command to install, i.e. conda install x
.
- PyTorch:
torch
(installation instructions)
- Note, make sure you install for your correct CPU/GPU version. This only really matters for matching cuda versions on the GPU, CPU install instructions should be straightforward.
- PyTorch Geometric and corresponding packages:
pyg
(installation instructions)
- Using
conda install pyg
should take care of version conflicts, so long as you first installtorch
- Rdkit:
rdkit
(installation instructions, see here also)
Additional pip requirements (latest version of each should work):
pandas
numpy
tqdm
(progress bar)matplotlib
sklearn
scipy
captum
(tools for explainability)networkx
Finally, you'll need to make sure that the reference to the polymerlearn
package is installed locally. From the main PolymerGNN directory, please run
pip install -e .