mirror of https://github.com/XingangPan/DragGAN
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
29 lines
689 B
Docker
29 lines
689 B
Docker
FROM nvcr.io/nvidia/pytorch:23.05-py3
|
|
|
|
ENV PYTHONDONTWRITEBYTECODE 1
|
|
ENV PYTHONUNBUFFERED 1
|
|
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
make \
|
|
pkgconf \
|
|
xz-utils \
|
|
xorg-dev \
|
|
libgl1-mesa-dev \
|
|
libglu1-mesa-dev \
|
|
libxrandr-dev \
|
|
libxinerama-dev \
|
|
libxcursor-dev \
|
|
libxi-dev \
|
|
libxxf86vm-dev \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
RUN pip install --no-cache-dir --upgrade pip
|
|
|
|
COPY requirements.txt .
|
|
RUN pip install --no-cache-dir -r requirements.txt
|
|
|
|
WORKDIR /workspace
|
|
|
|
RUN (printf '#!/bin/bash\nexec \"$@\"\n' >> /entry.sh) && chmod a+x /entry.sh
|
|
ENTRYPOINT ["/entry.sh"]
|