Qdqbert example add benchmark script with ORT-TRT (#16592)
* add ort-trt benchmark script * Update README.md * ort version can be newer * formatting * specify ORT version
This commit is contained in:
@@ -23,6 +23,7 @@ RUN python3 -m pip install --no-cache-dir --upgrade pip
|
||||
RUN python3 -m pip install --no-cache-dir --ignore-installed pycuda
|
||||
RUN python3 -m pip install --no-cache-dir \
|
||||
pytorch-quantization --extra-index-url https://pypi.ngc.nvidia.com
|
||||
RUN python3 -m pip install --no-cache-dir onnxruntime-gpu==1.11
|
||||
|
||||
WORKDIR /workspace
|
||||
COPY . transformers/
|
||||
|
||||
Reference in New Issue
Block a user