allencbzhang commited on
Commit
99f7158
·
verified ·
1 Parent(s): 91ef096

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- torch==2.2.0
2
  torchvision
3
  cloudpickle
4
  hydra-core
 
1
+ torch==2.2.0 --index-url https://download.pytorch.org/whl/cu118
2
  torchvision
3
  cloudpickle
4
  hydra-core