Skip to content

Commit 7675ac6

Browse files
committed
{nightly} Add nvshmem
1 parent f3c5b3a commit 7675ac6

File tree

1 file changed

+8
-7
lines changed

1 file changed

+8
-7
lines changed

nightly/Dockerfile

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -51,12 +51,14 @@ less \
5151
vim \
5252
ca-certificates \
5353
libjpeg-dev \
54-
libpng-dev \
55-
# This meta package will install libcusparselt0-cuda-13 & libcusparselt0-dev-cuda-13
54+
libpng-dev
55+
56+
RUN --mount=type=cache,target=/var/cache/apt \
57+
apt-get install -y --no-install-recommends \
58+
# Will also install libcusparselt0-cuda-13 & libcusparselt0-dev-cuda-13
5659
cusparselt-cuda-13 \
57-
# ffmpeg \
58-
# x264 \
59-
# x265 \
60+
# Will also install libnvshmem3-cuda-13 & libnvshmem3-dev-cuda-13 & libnvshmem3-static-cuda-13
61+
nvshmem-cuda-13 \
6062
&& rm -vrf /var/lib/apt/lists/* \
6163
&& rm -vf /usr/lib/python3.14/EXTERNALLY-MANAGED
6264

@@ -104,8 +106,7 @@ RUN git clone 'https://github.com/comfyanonymous/ComfyUI.git'
104106

105107
# Install only essential dependencies for ComfyUI
106108
RUN --mount=type=cache,target=/root/.cache/pip \
107-
awk '/^#non essential dependencies:/{exit} {print}' /root/ComfyUI/requirements.txt > /root/requirements-essential.txt \
108-
&& pip install -r /root/requirements-essential.txt
109+
pip install -r /root/ComfyUI/requirements.txt
109110

110111
################################################################################
111112
# Clean cache to avoid GitHub Actions "No space left on device"

0 commit comments

Comments
 (0)