Jingkang Yang commited on
Commit
249a0dd
·
1 Parent(s): 997d6df

update:requirments

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -0
requirements.txt CHANGED
@@ -11,5 +11,9 @@ imageio
11
  fvcore
12
  iopath
13
  pycocotools
 
 
 
 
14
  git+https://github.com/facebookresearch/detectron2.git
15
  git+https://github.com/openai/CLIP.git
 
11
  fvcore
12
  iopath
13
  pycocotools
14
+ pytorch==1.10.1
15
+ torchvision==0.11.2
16
+ torchaudio==0.10.1
17
+ cudatoolkit=11.3
18
  git+https://github.com/facebookresearch/detectron2.git
19
  git+https://github.com/openai/CLIP.git