xinjjj commited on
Commit
c806064
·
1 Parent(s): 55cded5
Files changed (1) hide show
  1. requirements.txt +8 -3
requirements.txt CHANGED
@@ -1,7 +1,12 @@
1
- --extra-index-url https://download.pytorch.org/whl/cu121
2
 
3
- torch==2.2.2
4
- torchvision==0.17.2
 
 
 
 
 
5
  pytorch-lightning==2.4.0
6
  lightning==2.3.3
7
  matplotlib
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu126
2
 
3
+ setuptools==80.10.2
4
+ wheel
5
+ packaging
6
+ Cython>=0.29.37
7
+
8
+ torch==2.8.0
9
+ torchvision==0.23.0
10
  pytorch-lightning==2.4.0
11
  lightning==2.3.3
12
  matplotlib