KingHacker9000 commited on
Commit
0e96d53
Β·
1 Parent(s): aaf29ec

Fix dependency conflicts (torch/xformers/cu118)

Browse files
Files changed (1) hide show
  1. requirements.txt +12 -11
requirements.txt CHANGED
@@ -1,4 +1,13 @@
 
 
1
  --extra-index-url https://download.pytorch.org/whl/cu118
 
 
 
 
 
 
 
2
  accelerate==1.0.1
3
  annotated-types==0.7.0
4
  anyio==4.5.2
@@ -19,7 +28,7 @@ easydict==1.13
19
  email_validator==2.2.0
20
  eval_type_backport==0.2.2
21
  exceptiongroup==1.3.0
22
- fastapi==0.115.1
23
  fastapi-cli==0.0.7
24
  filelock==3.16.1
25
  fonttools==4.57.0
@@ -63,13 +72,7 @@ mpmath==1.3.0
63
  networkx==3.1
64
  numba==0.58.1
65
  numpy==1.24.4
66
- nvidia-cudnn-cu12==9.1.0.70
67
- nvidia-cufft-cu12==11.0.2.54
68
- nvidia-curand-cu12==10.3.2.106
69
- nvidia-cusolver-cu12==11.4.5.107
70
- nvidia-cusparse-cu12==12.1.0.106
71
- nvidia-nvjitlink-cu12==12.9.86
72
- nvidia-nvtx-cu12==12.1.105
73
  opencv-python==4.5.4.60
74
  orjson==3.10.15
75
  packaging==25.0
@@ -117,8 +120,6 @@ sympy==1.13.3
117
  tifffile
118
  tokenizers==0.20.3
119
  toolz
120
- torch==2.3.0+cu118
121
- torchvision==0.18.0+cu118
122
  torch-tools==0.1.5
123
  tornado==6.4.2
124
  tqdm==4.67.1
@@ -136,5 +137,5 @@ watchfiles==0.24.0
136
  wcwidth==0.2.13
137
  websocket-client==1.8.0
138
  websockets==13.1
139
- xformers==0.0.27
140
  zipp==3.20.2
 
 
1
+ # ------------------------------------------------------------------
2
+ # Binary PyTorch wheels built for CUDA 11.8 (work on T4, A10G, A100)
3
  --extra-index-url https://download.pytorch.org/whl/cu118
4
+
5
+ torch==2.3.1+cu118
6
+ torchvision==0.18.1+cu118
7
+ xformers==0.0.27 # built against Torch 2.3.x
8
+ # torchaudio==2.3.1+cu118 # ← add only if you use it
9
+ # ------------------------------------------------------------------
10
+
11
  accelerate==1.0.1
12
  annotated-types==0.7.0
13
  anyio==4.5.2
 
28
  email_validator==2.2.0
29
  eval_type_backport==0.2.2
30
  exceptiongroup==1.3.0
31
+ fastapi==0.115.1 # pulls starlette 0.38.x automatically
32
  fastapi-cli==0.0.7
33
  filelock==3.16.1
34
  fonttools==4.57.0
 
72
  networkx==3.1
73
  numba==0.58.1
74
  numpy==1.24.4
75
+ # β€”β€” Removed all CUDA-12 wheels that conflict with cu118 β€”β€”
 
 
 
 
 
 
76
  opencv-python==4.5.4.60
77
  orjson==3.10.15
78
  packaging==25.0
 
120
  tifffile
121
  tokenizers==0.20.3
122
  toolz
 
 
123
  torch-tools==0.1.5
124
  tornado==6.4.2
125
  tqdm==4.67.1
 
137
  wcwidth==0.2.13
138
  websocket-client==1.8.0
139
  websockets==13.1
 
140
  zipp==3.20.2
141
+