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

71033a73 · Andrew Simonetta · c45dfd5a · 71033a73 · 71033a73
Commit 71033a73 authored Apr 21, 2021 by Andrew Simonetta
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

Dockerfile Dockerfile +1 -1

hardening_manifest.yaml hardening_manifest.yaml +1 -1

No files found.
--- a/Dockerfile
+++ b/Dockerfile
@@ -15,7 +15,7 @@ RUN dnf -y upgrade && dnf -y install pkg-config libgomp libpng python38 && dnf c
    echo "/usr/local/nvidia/lib64" >> /etc/ld.so.conf.d/nvidia.conf

 #ENV
-ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH}
+ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:/opt/tritonserver/bin:${PATH}
 ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64:/opt/tritonserver/lib:/opt/tritonserver/lib

 #Link

--- a/hardening_manifest.yaml
+++ b/hardening_manifest.yaml
@@ -42,7 +42,7 @@ resources:
    filename: "structure.tar.gz"
    validation:
      type: "sha256"
-      value: "11affeb49ce94e8c1be265e77fd5a0bac32c65534163010e670cf7644e8f5053"
+      value: "34c1d8aaedb45c26c63b883bc87d506533ac120287ad3de9719c59295937a0a8"

 maintainers:
  - name: "Daniel Mumpton"