wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-10 05:22:34 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

Added DockerFile

Browse Source
This commit is contained in:
jarvis2324 2023-07-23 18:13:32 +00:00
parent 30de083dd0
commit 782a9e29f9
1 changed files with 34 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
Dockerfile Normal file
View File

@ -0,0 +1,34 @@
# Use the official Python image as the base image
FROM nvidia/cuda:11.0.3-base-ubuntu20.04
# Run update
RUN apt update
#Install Python
RUN apt-get install -y python3 python3-pip
# Set the working directory inside the container
WORKDIR /app
# Copy the contents of the cloned GitHub repository to the working directory in the container
COPY ./ /app
# Install Python dependencies
RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118 xformers
RUN pip3 install -r requirements.txt
# Expose the port the application will be running on
EXPOSE 8188
# Set the environment variable for GPU support
ENV NVIDIA_VISIBLE_DEVICES all
# Run the Python program when the container starts
CMD ["python3", "main.py", "--listen","0.0.0.0"]