|
FROM nvidia/cuda:12.3.2-base-ubuntu22.04 |
|
LABEL authors="zxasv" |
|
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive |
|
|
|
|
|
RUN apt-get update && \ |
|
apt-get install -y \ |
|
git \ |
|
python3-pip \ |
|
python3-dev \ |
|
python3-opencv \ |
|
libglib2.0-0 |
|
|
|
COPY requirements.txt requirements.txt |
|
RUN ls -la / |
|
RUN python3 -m pip install --upgrade pip |
|
RUN python3 -m pip install --no-cache-dir -r requirements.txt |
|
|
|
RUN python3 -m pip install --upgrade pip |
|
|
|
|
|
RUN pip3 install torch torchvision torchaudio |
|
|
|
|
|
WORKDIR /app |
|
|
|
COPY / / |
|
RUN ls -la / |
|
|
|
|
|
ENTRYPOINT [ "python3" ] |
|
|
|
|
|
|