luodian commited on
Commit
2f32ba7
1 Parent(s): 4791b4d

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -0
requirements.txt CHANGED
@@ -1 +1,5 @@
 
 
 
 
1
  mmcv==2.0.0
 
1
+ torch==1.7.0
2
+ torchvision==0.8.0
3
+ torchaudio==0.7.0
4
+ cudatoolkit==10.1
5
  mmcv==2.0.0