IDM-VITON-MM / Dockerfile
parokshsaxena
updating dockerfile based on chatGPT for error module torch not found
b436f14
raw
history blame
1.08 kB
FROM continuumio/anaconda3:main
WORKDIR /code
COPY ./environment.yml /code/environment.yml
# Create the environment using the environment.yml file
RUN conda env create -f /code/environment.yml
# Make RUN commands use the new environment: From ChatGPT
SHELL ["conda", "run", "-n", "idm", "/bin/bash", "-c"]
# Set up a new user named "user" with user ID 1000
RUN useradd -m -u 1000 user
# Switch to the "user" user
USER user
# Set home to the user's home directory
ENV HOME=/home/user \
PYTHONPATH=$HOME/app \
PYTHONUNBUFFERED=1 \
GRADIO_ALLOW_FLAGGING=never \
GRADIO_NUM_PORTS=1 \
GRADIO_SERVER_NAME=0.0.0.0 \
GRADIO_THEME=huggingface \
SYSTEM=spaces
# Set the working directory to the user's home directory
WORKDIR $HOME/app
# Copy the current directory contents into the container at $HOME/app setting the owner to the user
COPY --chown=user . $HOME/app
# From ChatGPT
# CMD ["./run.sh"]
# Make sure conda is initialized for the user
RUN echo "conda activate idm" >> ~/.bashrc
CMD ["conda", "run", "--no-capture-output", "-n", "idm", "./run.sh"]