mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-06 00:03:22 +00:00
6dfdef7782
### Description Update docker and package version for stable diffusion demo. ### Motivation and Context Update onnx to 1.16 for security
22 lines
825 B
Text
22 lines
825 B
Text
-r requirements.txt
|
|
|
|
# Official onnxruntime-gpu 1.16.1 is built with CUDA 11.8.
|
|
onnxruntime-gpu>=1.16.2
|
|
|
|
py3nvml
|
|
|
|
# The version of cuda-python shall be compatible with installed CUDA version.
|
|
# For demo of TensorRT excution provider and TensortRT.
|
|
cuda-python==11.8.0
|
|
|
|
# For windows, cuda-python need the following
|
|
pywin32; platform_system == "Windows"
|
|
|
|
# For windows, run `conda install -c conda-forge nvtx` instead
|
|
nvtx; platform_system != "Windows"
|
|
|
|
# Please install PyTorch 2.1 or above for CUDA 11.8 using one of the following commands:
|
|
# pip3 install torch --index-url https://download.pytorch.org/whl/cu118
|
|
|
|
# Run the following command to install some extra packages for onnx graph optimization for TensorRT manually.
|
|
# pip3 install --upgrade polygraphy onnx-graphsurgeon --extra-index-url https://pypi.ngc.nvidia.com
|