requirements-inference-onnxruntime-gpu.txt 242 B

123456
  1. # 与 CUDA 匹配的 GPU 版 ONNX Runtime;安装前请对照 NVIDIA 驱动与
  2. # https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html
  3. # 再选具体版本,例如:
  4. # onnxruntime-gpu>=1.16.0
  5. onnxruntime-gpu>=1.16.0