Docker: fix missing libnvinfer7

ertugrul-dmr · May 1, 2023 · 1daafcc · 1daafcc
1 parent 6a7ca53
commit 1daafcc
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -24,10 +24,14 @@ RUN . ./venv/bin/activate && \
     python3 -m pip install --use-pep517 --no-deps -U triton==2.0.0 torch>=2.0.0+cu121 xformers \
 	                       --extra-index-url https://download.pytorch.org/whl/cu121
 
+# Fix missing libnvinfer7
+USER root
+RUN ln -s /usr/lib/x86_64-linux-gnu/libnvinfer.so /usr/lib/x86_64-linux-gnu/libnvinfer.so.7 && \
+    ln -s /usr/lib/x86_64-linux-gnu/libnvinfer_plugin.so /usr/lib/x86_64-linux-gnu/libnvinfer_plugin.so.7
+
 USER appuser
 COPY --chown=appuser . .
 
-# https://github.com/kohya-ss/sd-scripts/issues/405#issuecomment-1509851709
 RUN sed -i 's/import library.huggingface_util/# import library.huggingface_util/g' train_network.py && \
     sed -i 's/import library.huggingface_util/# import library.huggingface_util/g' library/train_util.py
 

diff --git a/docker-compose.yaml b/docker-compose.yaml
@@ -7,7 +7,6 @@ services:
       context: .
     ports:
       - 127.0.0.1:7860:7860
-    stdin_open: true
     tty: true
     ipc: host
     environment: