Skip to content

Commit ba2840d

Browse files
rajeevsraorajeevsrao
committed
Update TensorRT Dockerfile
1 parent af69862 commit ba2840d

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

‎TensorFlow/LanguageModeling/BERT/trt/Dockerfile‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -58,8 +58,8 @@ COPY requirements.txt /tmp/requirements.txt
5858
RUN python3 -m pip install -r /tmp/requirements.txt
5959

6060
# Build with extra flags to enable plugins available on SM 70/75
61-
RUN git clone https://github.com/rajeevsrao/TensorRT.git
62-
RUN cd TensorRT && git checkout origin/dev/release/7.0 && git submodule update --init --recursive && mkdir build && cd build && cmake .. -DGPU_ARCHS="70 75" -DBUILD_PARSERS=OFF && make -j nvinfer_plugin
61+
RUN git clone https://github.com/NVIDIA/TensorRT.git
62+
RUN cd TensorRT && git checkout 9b2d78a87e3d45eca22b6bfcb6e10777b932ef43 && git submodule update --init --recursive && mkdir build && cd build && cmake .. -DGPU_ARCHS="70 75" -DBUILD_PARSERS=OFF && make -j nvinfer_plugin
6363
# Replace the default plugin library with the open-source plugins
6464
RUN cp TensorRT/build/libnvinfer_plugin.so* /usr/lib/x86_64-linux-gnu/
6565

0 commit comments

Comments
 (0)