Spaces:
Runtime error
Runtime error
GlandVergil
commited on
Update requirements.txt
Browse files- 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
|