Spaces:
Running
on
L40S
Running
on
L40S
Commit
•
d249624
1
Parent(s):
3886235
add doc
Browse files- requirements.txt +4 -0
requirements.txt
CHANGED
|
@@ -36,7 +36,11 @@ ffmpeg-python==0.2.0
|
|
| 36 |
|
| 37 |
# versions of onnx gpu <= 1.17 did not support Cuda 12
|
| 38 |
onnxruntime-gpu==1.19.2
|
|
|
|
|
|
|
|
|
|
| 39 |
onnx==1.16.2
|
|
|
|
| 40 |
scikit-image==0.20.0
|
| 41 |
albumentations==1.3.1
|
| 42 |
matplotlib==3.7.2
|
|
|
|
| 36 |
|
| 37 |
# versions of onnx gpu <= 1.17 did not support Cuda 12
|
| 38 |
onnxruntime-gpu==1.19.2
|
| 39 |
+
|
| 40 |
+
# onnx 1.16.2 has some issues it seems:
|
| 41 |
+
# https://github.com/onnx/onnx/issues/6267
|
| 42 |
onnx==1.16.2
|
| 43 |
+
|
| 44 |
scikit-image==0.20.0
|
| 45 |
albumentations==1.3.1
|
| 46 |
matplotlib==3.7.2
|