Skip to content

Commit

Permalink
updated req files for dc280
Browse files Browse the repository at this point in the history
  • Loading branch information
mauvais2 committed Sep 25, 2024
1 parent f9efd21 commit 04a7e13
Show file tree
Hide file tree
Showing 4 changed files with 11 additions and 8 deletions.
2 changes: 1 addition & 1 deletion pip/cpu_requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@

tensorflow-cpu~=2.14.0

torch==2.2.0+cpu
torch==2.2.1+cpu
torch_geometric
lightning

Expand Down
8 changes: 5 additions & 3 deletions pip/cuda_requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -4,12 +4,13 @@
# **Note**: for LC developers, comment out line 3 pypi install and use/uncommenting line 5:
#--extra-index-url https://wci-repo.llnl.gov/repository/pypi-group/simple
-f https://data.dgl.ai/wheels/cu118/repo.html
-f https://download.pytorch.org/whl/cu118/torch_stable.html

tensorflow[and-cuda]~=2.14.0
tensorflow[and-cuda]~=2.15.0

tensorrt

torch==2.2.0
torch==2.2.1+cu118
torch_geometric
lightning

Expand Down Expand Up @@ -46,5 +47,6 @@ ipykernel
deepchem==2.8.0
rdkit==2024.3.5

# deepchem 2.8 requires lightgbm. lightgbm
# deepchem 2.8 requires lightgbm.
lightgbm
pyyaml==5.4.1
5 changes: 3 additions & 2 deletions pip/docker_requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@

tensorflow-cpu~=2.14.0

torch==2.2.0+cpu
torch==2.2.1+cpu
torch_geometric
lightning

Expand Down Expand Up @@ -46,5 +46,6 @@ jupyterlab
deepchem==2.8.0
rdkit==2024.3.5

# deepchem 2.8 requires lightgbm. lightgbm
# deepchem 2.8 requires lightgbm.
lightgbm
pyyaml==5.4.1
4 changes: 2 additions & 2 deletions pip/mchip_requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
# use tensorflow not tensorflow-cpu for arm. see https://www.tensorflow.org/install/pip#linux
tensorflow~=2.14.0

torch==2.2.0+cpu
torch==2.2.1+cpu
torch_geometric
lightning

Expand Down Expand Up @@ -44,7 +44,7 @@ ipykernel
notebook
jupyterlab

deepchem==2.7.1
deepchem==2.8.0
rdkit==2024.3.5

# deepchem requires lightgbm. need to check if it works for macOS. https://pypi.org/project/lightgbm/
Expand Down

0 comments on commit 04a7e13

Please sign in to comment.