Skip to content

Commit

Permalink
Minor updates to requirements.txt
Browse files Browse the repository at this point in the history
  • Loading branch information
jacobdurrant committed Dec 19, 2023
1 parent 6a8903c commit 78307f8
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 2 deletions.
5 changes: 4 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -209,9 +209,12 @@ You can build a virtualenv with the requirements:
$ python3 -m venv leadopt_env
$ source ./leadopt_env/bin/activate
$ pip install -r requirements.txt
$ pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html
$ pip install torch==2.1.2+cu118 torchvision==0.16.2+cu118 --index-url https://download.pytorch.org/whl/cu118
```

<!-- pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html -->
<!-- 2.1.2+cu118 -->

Note: `Cuda 10.1` is required during training

## Training
Expand Down
3 changes: 2 additions & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -30,4 +30,5 @@ tqdm==4.48.2
urllib3==1.25.10
wandb==0.9.4
watchdog==0.10.3
rdkit==2023.09.2
rdkit==2023.09.2
prody==2.4.1

0 comments on commit 78307f8

Please sign in to comment.