onnxruntime/onnxruntime/python/tools/transformers/requirements_cpu.txt

14 lines
228 B
Text

onnx
numpy
coloredlogs
psutil
py-cpuinfo
py3nvml
packaging
transformers
onnxruntime
onnxconverter_common
--find-links https://download.pytorch.org/whl/torch_stable.html
torch==1.7.1+cpu
torchvision==0.8.2+cpu
torchaudio===0.7.2