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

Updated Dockerfile

Browse Source
This commit is contained in:
root 2023-08-07 19:22:22 +00:00
parent 2c933106e5
commit 92b03a6875
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
Dockerfile
View File

@ -1,7 +1,6 @@
# Use the official Python image as the base image # Use the official Python image as the base image
FROM nvidia/cuda:11.0.3-base-ubuntu20.04 FROM nvidia/cuda:11.0.3-base-ubuntu20.04
# Run update # Run update
RUN apt update RUN apt update
@ -23,6 +22,8 @@ RUN pip3 install torch torchvision torchaudio --extra-index-url https://download
RUN pip3 install -r requirements.txt RUN pip3 install -r requirements.txt
RUN pip3 install runpod RUN pip3 install runpod
RUN pip3 install websocket-client RUN pip3 install websocket-client
RUN pip3 install webcolors
RUN pip3 install scikit-learn
#Give permission to script #Give permission to script
RUN chmod +x ./entrypoint.sh RUN chmod +x ./entrypoint.sh