Pinwheel commited on
Commit
7b7cd62
1 Parent(s): ca2b668

Fix torch version

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -2
requirements.txt CHANGED
@@ -1,3 +1,6 @@
 
 
 
1
  absl-py==1.2.0
2
  aiohttp==3.8.3
3
  aiosignal==1.2.0
@@ -133,8 +136,6 @@ timm==0.6.7
133
  tokenizers==0.10.3
134
  toml==0.10.2
135
  toolz==0.12.0
136
- torch==1.10.0
137
- torchvision==0.11.0
138
  tornado==6.2
139
  tqdm==4.64.1
140
  traitlets==5.4.0
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu113
2
+ torch==1.10.0+cu113
3
+ torchvision==0.11.0+cu113
4
  absl-py==1.2.0
5
  aiohttp==3.8.3
6
  aiosignal==1.2.0
 
136
  tokenizers==0.10.3
137
  toml==0.10.2
138
  toolz==0.12.0
 
 
139
  tornado==6.2
140
  tqdm==4.64.1
141
  traitlets==5.4.0