diff --git a/requirements_cpu.txt b/requirements_cpu.txt new file mode 100644 index 0000000..40f457d --- /dev/null +++ b/requirements_cpu.txt @@ -0,0 +1,3 @@ +numpy==1.21.6 +onnx==1.13.1 +onnxruntime==1.14.0 \ No newline at end of file diff --git a/requirements_gpu.txt b/requirements_gpu.txt new file mode 100644 index 0000000..b0675c0 --- /dev/null +++ b/requirements_gpu.txt @@ -0,0 +1,3 @@ +numpy==1.21.6 +onnx==1.12.0 +onnxruntime-gpu==1.14.0 \ No newline at end of file