update base cuda version to 11.8; some dockerfile-gpu fixes

This commit is contained in:
makaveli10
2024-02-19 04:38:41 -05:00
parent ff871ad485
commit d1de4948ee
+9 -23
View File
@@ -1,44 +1,30 @@
FROM nvidia/cuda:11.2.2-cudnn8-runtime-ubuntu20.04 FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
ARG DEBIAN_FRONTEND=noninteractive ARG DEBIAN_FRONTEND=noninteractive
# Remove any third-party apt sources to avoid issues with expiring keys. # Remove any third-party apt sources to avoid issues with expiring keys.
RUN rm -f /etc/apt/sources.list.d/*.list RUN rm -f /etc/apt/sources.list.d/*.list
# Install some basic utilities. # Install some basic utilities.
RUN apt-get update && apt-get install -y \ RUN apt-get update && apt-get install -y --no-install-recommends \
curl \ curl \
ca-certificates \ ca-certificates \
sudo \ sudo \
git \ git \
bzip2 \ bzip2 \
libx11-6 \ libx11-6 \
&& rm -rf /var/lib/apt/lists/* python3-dev \
python3-pip \
RUN apt update && python3 -m pip install --upgrade pip \
&& rm -rf /var/lib/apt/lists/*
# install python
RUN apt install software-properties-common -y && \
add-apt-repository ppa:deadsnakes/ppa && \
apt update
RUN apt install python3-dev -y && \
apt install python-is-python3
# install pip
RUN apt install python3-pip -y
# Create a working directory. # Create a working directory.
RUN mkdir /app RUN mkdir /app
WORKDIR /app WORKDIR /app
COPY scripts/setup.sh /app COPY scripts/setup.sh requirements/server.txt /app
COPY requirements/ /app
RUN apt update --fix-missing RUN apt update && bash setup.sh && rm setup.sh
RUN bash setup.sh RUN pip install -r server.txt && rm server.txt
RUN pip install -r server.txt
COPY whisper_live /app/whisper_live COPY whisper_live /app/whisper_live