veb-101 commited on
Commit
cc6b9d8
·
1 Parent(s): 5c6bb4d

updated torch package finding location

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -2
requirements.txt CHANGED
@@ -1,4 +1,5 @@
1
- torch==2.0.0
2
- torchvision
 
3
  lightning==2.0.1
4
  torchmetrics==1.0.0
 
1
+ --find-links https://download.pytorch.org/whl/torch_stable.html
2
+ torch==2.0.0+cpu
3
+ torchvision==0.15.0
4
  lightning==2.0.1
5
  torchmetrics==1.0.0