Update requirements.txt

2026-04-10 15:00:23 +00:00 · 2024-03-18 22:31:59 +08:00
parent 5410a8c79b
commit a71c35ccd9
1 changed files with 3 additions and 0 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -25,6 +25,9 @@ huggingface-hub==0.20.1
 # onnx==1.15.1
 # onnxruntime-gpu==1.17.1
 # onnxruntime==1.17.1
+# for cuda 12.1(default 11.8)
+# onnxruntime-gpu --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
+
 # this is for onnx: 
 # protobuf==3.20.3
 # open clip for SDXL