bennyguo commited on
Commit
d286ae5
·
1 Parent(s): 009cfd7

update torch version

Browse files
Files changed (1) hide show
  1. pre-requirements.txt +3 -2
pre-requirements.txt CHANGED
@@ -1,4 +1,5 @@
1
- torch
2
- torchvision
3
  ninja
 
 
 
4
 
 
 
 
1
  ninja
2
+ --extra-index-url https://download.pytorch.org/whl/cu113
3
+ torch==1.12.1+cu113
4
+ torchvision==0.13.1+cu113
5