Fixed Nvidia Docker dependencies (#1327)

1. The update to Streamlit 1.13 has bee mirrored
2. Streamlit needs to be in Headless mode so it doesn't pester about
wanting an email on startup, which crashes in docker since there's no
stdin
3. Fixed an issue where opencv-python from the base image was not
allowing us to override it with opencv-python-headless, which is
required because of its lack of runtime DLLs (+ it's smaller)
This commit is contained in:
Travis Fletcher 2022-09-26 17:42:02 -04:00 committed by GitHub
parent b1f125ea06
commit f6394a89ca
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 6 additions and 3 deletions

View File

@ -23,7 +23,7 @@ fastReruns = false
folderWatchBlacklist = []
fileWatcherType = "auto"
cookieSecret = ""
headless = false
headless = true
runOnSave = false
port = 8501
baseUrlPath = ""

View File

@ -31,7 +31,10 @@ RUN apt-get update && \
rm -rf /var/lib/apt/lists/*
COPY ./requirements.txt /install/
RUN pip install -r /install/requirements.txt
# From base image. We need opencv-python-headless so we uninstall here
RUN pip uninstall -y opencv-python && pip install opencv-python-headless==4.6.0.66
# Install font for prompt matrix
COPY /data/DejaVuSans.ttf /usr/share/fonts/truetype/

View File

@ -16,7 +16,7 @@ pytorch_lightning==1.7.6
# TODO: Pin external dependency versions
#opencv-python==4.6.0.66 # Opencv python already satisfied upstream
opencv-python-headless==4.6.0.66 # Needed to operate opencv in headless/server mode
#opencv-python-headless==4.6.0.66 # Needed to operate opencv in headless/server mode
taming-transformers-rom1504==0.0.6 # required by ldm
@ -43,7 +43,7 @@ gradio==3.3.1
# Environment Dependencies for WebUI (streamlit)
streamlit==1.12.2
streamlit==1.13.0
streamlit-on-Hover-tabs==1.0.1
streamlit-option-menu==0.3.2
streamlit_nested_layout==0.1.1