Spaces:
Running
Running
File size: 684 Bytes
e3598ee 78dd8f2 e3598ee 78dd8f2 e3598ee |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
# Use an official conda-based Python image as a parent image
FROM pytorch/pytorch:2.2.1-cuda12.1-cudnn8-runtime
LABEL maintainer vincentqyw
ARG PYTHON_VERSION=3.10.10
# Set the working directory to /code
WORKDIR /code
# Copy the current directory contents into the container at /code
COPY . /code
RUN conda create -n imw python=${PYTHON_VERSION}
RUN echo "source activate imw" > ~/.bashrc
ENV PATH /opt/conda/envs/imw/bin:$PATH
# Make RUN commands use the new environment
SHELL ["conda", "run", "-n", "imw", "/bin/bash", "-c"]
RUN pip install --upgrade pip
RUN pip install -r env-docker.txt
RUN apt-get update && apt-get install ffmpeg libsm6 libxext6 -y
# Export port
EXPOSE 7860
|