Commit 1097a8ad authored by VIGNESHinZONE's avatar VIGNESHinZONE
Browse files

Merge branch 'set-torch' into new-setup

parents 321134a9 f55f4d4b
Loading
Loading
Loading
Loading
+15 −0
Original line number Diff line number Diff line
name: deepchem
channels:
  - omnia
  - conda-forge
  - defaults
dependencies:
  - numpy==1.20.*
  - pip==20.2.*
  - pip:
    - tensorflow~=2.4
    - dgllife==0.2.8
    - pymatgen==2020.12.31
    - matminer
    - networkx
    - torch-geometric==1.7.0
+2 −1
Original line number Diff line number Diff line
@@ -11,7 +11,8 @@ else:

# Environment-specific dependencies.
extras = {
    'jax': ['jax==0.2.14', 'jaxlib==0.1.67', 'dm-haiku==0.0.4', 'optax==0.0.8']
    'jax': ['jax==0.2.14', 'jaxlib==0.1.67', 'dm-haiku==0.0.4', 'optax==0.0.8'],
    'torch': ['torch', 'torchvision', 'dgl', 'dgllife']
}