onnxruntime/onnxruntime/python/tools/transformers/models/llama/requirements-cuda.txt
ironman 9acaf534a6
Benchmark - Updating llama-2 requirement files (#19716)
### Description
<!-- Describe your changes. -->



### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->
2024-03-04 07:29:58 -08:00

5 lines
No EOL
230 B
Text

-r requirements.txt
# Please manually install torch>=2.2.0 with CUDA enabled for the CUDA version installed in your system.
# Instructions can be found here: https://pytorch.org/get-started/locally/
onnxruntime-gpu>=1.16.2
py3nvml