Loading docs/requirements.txt +1 −1 Original line number Diff line number Diff line Loading @@ -4,4 +4,4 @@ sphinx>=3.2,<4 sphinx_rtd_theme>=0.5,<1 tensorflow>=2.3,<2.4 transformers torch==1.6.0 torch==1.8.1 env.common.yml +1 −2 Original line number Diff line number Diff line Loading @@ -25,8 +25,7 @@ dependencies: - tensorflow~=2.4 - tensorflow_probability==0.12.* - tensorflow_addons==0.12.* - torch-geometric==1.6.* - torch-geometric==1.7.0 - transformers==3.5.* - xgboost==1.* - git+https://github.com/samoturk/mol2vec - git+https://github.com/deepmind/dm-haiku env.cpu.mac.yml +4 −6 Original line number Diff line number Diff line dependencies: - pip: - -f https://pytorch-geometric.com/whl/torch-1.6.0.html - -f https://pytorch-geometric.com/whl/torch-1.6.0+cpu.html - -f https://pytorch-geometric.com/whl/torch-1.8.1.html - -f https://pytorch-geometric.com/whl/torch-1.8.0+cpu.html - dgl==0.6.* - torch==1.6.0 - torchvision==0.7.0 - torch==1.8.1 - torchvision==0.9.1 - torch-cluster - torch-scatter - torch-sparse - torch-spline-conv==1.2.0 env.cpu.yml +3 −5 Original line number Diff line number Diff line dependencies: - pip: - -f https://download.pytorch.org/whl/torch_stable.html - -f https://pytorch-geometric.com/whl/torch-1.6.0+cpu.html - -f https://pytorch-geometric.com/whl/torch-1.8.0+cpu.html - dgl==0.6.* - torch==1.6.0+cpu - torchvision==0.7.0+cpu - torch==1.8.1+cpu - torchvision==0.9.1+cpu - torch-cluster - torch-scatter - torch-sparse - torch-spline-conv==1.2.0 env.gpu.yml +4 −6 Original line number Diff line number Diff line dependencies: - pip: - -f https://download.pytorch.org/whl/torch_stable.html - -f https://pytorch-geometric.com/whl/torch-1.6.0+cu101.html - dgl-cu101==0.6.* - torch==1.6.0+cu101 - torchvision==0.7.0+cu101 - -f https://pytorch-geometric.com/whl/torch-1.8.1+cu111.html - dgl-cu110==0.6.* - torch==1.8.1+cu111 - torchvision==0.9.1+cu111 - torch-cluster - torch-scatter - torch-sparse - torch-spline-conv==1.2.0 Loading
docs/requirements.txt +1 −1 Original line number Diff line number Diff line Loading @@ -4,4 +4,4 @@ sphinx>=3.2,<4 sphinx_rtd_theme>=0.5,<1 tensorflow>=2.3,<2.4 transformers torch==1.6.0 torch==1.8.1
env.common.yml +1 −2 Original line number Diff line number Diff line Loading @@ -25,8 +25,7 @@ dependencies: - tensorflow~=2.4 - tensorflow_probability==0.12.* - tensorflow_addons==0.12.* - torch-geometric==1.6.* - torch-geometric==1.7.0 - transformers==3.5.* - xgboost==1.* - git+https://github.com/samoturk/mol2vec - git+https://github.com/deepmind/dm-haiku
env.cpu.mac.yml +4 −6 Original line number Diff line number Diff line dependencies: - pip: - -f https://pytorch-geometric.com/whl/torch-1.6.0.html - -f https://pytorch-geometric.com/whl/torch-1.6.0+cpu.html - -f https://pytorch-geometric.com/whl/torch-1.8.1.html - -f https://pytorch-geometric.com/whl/torch-1.8.0+cpu.html - dgl==0.6.* - torch==1.6.0 - torchvision==0.7.0 - torch==1.8.1 - torchvision==0.9.1 - torch-cluster - torch-scatter - torch-sparse - torch-spline-conv==1.2.0
env.cpu.yml +3 −5 Original line number Diff line number Diff line dependencies: - pip: - -f https://download.pytorch.org/whl/torch_stable.html - -f https://pytorch-geometric.com/whl/torch-1.6.0+cpu.html - -f https://pytorch-geometric.com/whl/torch-1.8.0+cpu.html - dgl==0.6.* - torch==1.6.0+cpu - torchvision==0.7.0+cpu - torch==1.8.1+cpu - torchvision==0.9.1+cpu - torch-cluster - torch-scatter - torch-sparse - torch-spline-conv==1.2.0
env.gpu.yml +4 −6 Original line number Diff line number Diff line dependencies: - pip: - -f https://download.pytorch.org/whl/torch_stable.html - -f https://pytorch-geometric.com/whl/torch-1.6.0+cu101.html - dgl-cu101==0.6.* - torch==1.6.0+cu101 - torchvision==0.7.0+cu101 - -f https://pytorch-geometric.com/whl/torch-1.8.1+cu111.html - dgl-cu110==0.6.* - torch==1.8.1+cu111 - torchvision==0.9.1+cu111 - torch-cluster - torch-scatter - torch-sparse - torch-spline-conv==1.2.0
