feng2022 commited on
Commit
081d122
1 Parent(s): f8af932

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -3
requirements.txt CHANGED
@@ -1,11 +1,13 @@
1
  # Torch
2
  #--find-links https://download.pytorch.org/whl/torch_stable.html
3
- torch==1.4.0+cu100
4
- torchvision==0.11.2+cu100
5
- torchaudio==0.10.1+cu100
6
  setuptools==59.5.0
7
  transformers==4.3.3
8
 
 
 
9
  Pillow
10
  ninja
11
  tqdm
 
1
  # Torch
2
  #--find-links https://download.pytorch.org/whl/torch_stable.html
3
+ torch==1.11.0
4
+ torchvision==0.12.0
5
+ torchaudio==0.11.0
6
  setuptools==59.5.0
7
  transformers==4.3.3
8
 
9
+ CUDA==10.1
10
+
11
  Pillow
12
  ninja
13
  tqdm