Stanislaw Szymanowicz commited on
Commit
693a19d
1 Parent(s): 4237c2d

Install cuda-compatible torch

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