Remove onnx graphsurgery from Dockerfile (NVIDIA#5320)

Signed-off-by: smajumdar <[email protected]> Signed-off-by: smajumdar <[email protected]> Signed-off-by: Hainan Xu <[email protected]>
hainan-xv · Nov 29, 2022 · 7966e2d · 7966e2d
1 parent 4168027
commit 7966e2d
Showing 1 changed file with 0 additions and 8 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -46,14 +46,6 @@ WORKDIR /tmp/torchaudio_build
 COPY scripts/installers /tmp/torchaudio_build/scripts/installers/
 RUN /bin/bash /tmp/torchaudio_build/scripts/installers/install_torchaudio_latest.sh
 
-#install TRT tools: PT quantization support and ONNX graph optimizer
-WORKDIR /tmp/trt_build
-RUN git clone https://github.com/NVIDIA/TensorRT.git && \
-    cd TensorRT/tools/onnx-graphsurgeon && python setup.py install && \
-    cd ../pytorch-quantization && \
-    python setup.py install && \
-    rm -fr  /tmp/trt_build
-
 # install nemo dependencies
 WORKDIR /tmp/nemo
 COPY requirements .