298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-07-29 02:47:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

changed dockerfile to include amd's gpu and pytorch ver.

Browse Source
This commit is contained in:
RedsAnalysis 2025-03-17 06:48:58 -04:00
parent b9c927d351
commit a52ecc6ea5
1 changed files with 28 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
Dockerfile
View File

@ -1,4 +1,18 @@
# Nvidia GPU Base Images
# For NVIDIA GPU with stable CUDA version
FROM nvidia/cuda:12.6.3-cudnn-runtime-ubuntu24.04 AS base
# For NVIDIA GPU with latest CUDA version
FROM nvidia/cuda:12.8.1-cudnn-runtime-ubuntu24.04 AS base
# AMD GPU Base Images
# For AMD GPU with stable ROCm version
FROM rocm/dev-ubuntu-24.04:6.2.4-complete AS base
# For AMD GPU with latest ROCm version
FROM rocm/dev-ubuntu-24.04:6.3.4-complete AS base
# Environment variables
ENV DEBIAN_FRONTEND=noninteractive
# Install necessary dependencies and Python 3.12
@ -29,15 +43,26 @@ RUN git clone https://github.com/comfyanonymous/ComfyUI.git /app/comfyui \
&& /app/venv/bin/pip install pyyaml \
&& /app/venv/bin/pip install -r /app/comfyui/requirements.txt
# Install PyTorch with CUDA 12.6 support (stable version)
RUN /app/venv/bin/pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu126
# Clone ComfyUI-Manager and install its dependencies
RUN git clone https://github.com/ltdrdata/ComfyUI-Manager.git /app/temp/ComfyUI-Manager \
&& mv /app/temp/* /app/comfyui/custom_nodes/ \
&& rm -rf /app/temp \
&& /app/venv/bin/pip install -r /app/comfyui/custom_nodes/ComfyUI-Manager/requirements.txt
# NVIDIA GPU PyTorch Installation
# Install PyTorch with CUDA 12.6 support (stable version)
RUN /app/venv/bin/pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu126
# Install PyTorch with CUDA 12.8 support (latest version)
RUN /app/venv/bin/pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128
# AMD GPU PyTorch Installation
# Install PyTorch with ROCm 6.2 support (stable version)
RUN /app/venv/bin/pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.2.4
# Install PyTorch with ROCm 6.3 support (latest version)
RUN /app/venv/bin/pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3
# Expose the backend port
EXPOSE 8188