# Use an official PyTorch runtime as a parent image
FROM pytorch/pytorch

RUN apt update && apt install -y ffmpeg libsm6 libxext6 libgl1*

# Set the working directory
WORKDIR /root/SEAC_Pytorch_Release/

# Copy the current directory contents into the container
COPY . /root/SEAC_Pytorch_release/
# Install any needed packages specified in requirements.txt
RUN pip3 install -r requirement.txt

# Run when the container launches
CMD ["python", "main.py"]
