UNCLASSIFIED

Commit d63944f9 authored by Andrew Simonetta's avatar Andrew Simonetta
Browse files

Added support for CPU only mode (w/ split layers)

parent 9f359f2d
Pipeline #228371 passed with stages
in 60 minutes and 46 seconds
......@@ -16,7 +16,7 @@ RUN dnf -y upgrade && dnf -y install pkg-config libgomp libpng python38 && dnf c
#ENV
ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH}
ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64:/opt/tritonserver/lib:/usr/lib64
ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64:/opt/tritonserver/lib:/opt/tritonserver/lib
#Link
RUN ln -s cuda-11.2 /usr/local/cuda
......@@ -54,6 +54,6 @@ EXPOSE 8000
EXPOSE 8001
EXPOSE 8002
#
ENTRYPOINT ["/opt/tritonserver/bin/tritonserver", "--strict-model-config=false", "--model-repository", "/models"]
ENTRYPOINT ["/opt/tritonserver/nvidia_entrypoint.sh"]
HEALTHCHECK --timeout=30s CMD curl -fs http://localhost:8000/v2 || exit 1
\ No newline at end of file
......@@ -42,7 +42,7 @@ resources:
filename: "structure.tar.gz"
validation:
type: "sha256"
value: "229416ebf2c8b7f4a534b2ee70d38c878a66cc80554d022d8a0a38b872f870fa"
value: "ab61aeea73ace268d8587cbb287e8d8c2462bb995fefd78fe40eac43923810a5"
maintainers:
- name: "Daniel Mumpton"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment