fix: always install OpenBLAS and Stable Diffusion requirements

This commit is contained in:
Sébastien Prud'homme 2023-05-29 21:14:09 +02:00
parent a7fe65a54c
commit 65809f303e
2 changed files with 17 additions and 20 deletions

View file

@ -19,6 +19,7 @@ WORKDIR /build
RUN apt-get update && \
apt-get install -y ca-certificates cmake curl
# CuBLAS requirements
RUN if [ "${BUILD_TYPE}" = "cublas" ]; then \
apt-get install -y software-properties-common && \
apt-add-repository contrib && \
@ -30,14 +31,12 @@ RUN if [ "${BUILD_TYPE}" = "cublas" ]; then \
; fi
ENV PATH /usr/local/cuda/bin:${PATH}
RUN if [ "${BUILD_TYPE}" = "openblas" ]; then \
apt-get install -y libopenblas-dev \
; fi
# OpenBLAS requirements
RUN apt-get install -y libopenblas-dev
RUN if [ "${GO_TAGS}" = "stablediffusion" ]; then \
apt-get install -y libopencv-dev && \
ln -s /usr/include/opencv4/opencv2 /usr/include/opencv2 \
; fi
# Stable Diffusion requirements
RUN apt-get install -y libopencv-dev && \
ln -s /usr/include/opencv4/opencv2 /usr/include/opencv2
COPY . .
RUN make build