GlandVergil commited on
Commit
9720c62
·
verified ·
1 Parent(s): b158c5a

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -2
requirements.txt CHANGED
@@ -7,8 +7,10 @@ decorator==5.1.0
7
  OmegaConf
8
  hydra-core
9
  hydra-py
10
- dgl
11
  torchaudio
12
  torchvision
13
  pyrsistent
14
- pytorch==1.9
 
 
 
7
  OmegaConf
8
  hydra-core
9
  hydra-py
10
+ dgl-cuda11.1
11
  torchaudio
12
  torchvision
13
  pyrsistent
14
+ pytorch==1.9
15
+ cudatoolkit==11.1
16
+ pip