Skip to content

Commit 737b025

Browse files
authored
Change the install cuda version in README
1 parent a13bbaa commit 737b025

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,9 +19,9 @@ Install pytorch with the cuda version that fits your device.<br>
1919
cd LocalTransform
2020
conda create -c conda-forge -n rdenv python=3.6 -y
2121
conda activate rdenv
22-
conda install pytorch==1.0.0 cuda80 -c pytorch -y
22+
conda install pytorch cudatoolkit=11.3 -c pytorch -y
2323
conda install -c conda-forge rdkit -y
24-
pip install dgl
24+
conda install -c dglteam dgl-cuda11.3
2525
pip install dgllife
2626
```
2727

0 commit comments

Comments
 (0)