Commit
Β·
0e96d53
1
Parent(s):
aaf29ec
Fix dependency conflicts (torch/xformers/cu118)
Browse files- 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 |
-
|
| 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 |
+
|