Skip to content

Commit f3c5b3a

Browse files
committed
{nightly} Add cusparselt
1 parent 2e356da commit f3c5b3a

File tree

1 file changed

+11
-9
lines changed

1 file changed

+11
-9
lines changed

nightly/Dockerfile

Lines changed: 11 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
################################################################################
22
# Dockerfile that builds 'yanwk/comfyui-boot:nightly'
3-
# A image with preview (nightly) dependencies for ComfyUI.
4-
# Does not download ComfyUI or any models.
3+
# A image for testing with latest ComfyUI and nightly/experimental dependencies.
54
#
65
# Using image from NVIDIA as base image:
76
# https://catalog.ngc.nvidia.com/orgs/nvidia/containers/cuda/tags
@@ -12,9 +11,9 @@
1211
# Note about GCC:
1312
# Ubuntu 24.04 comes with GCC 13, so no additional config here.
1413
#
15-
# Note about environment variables:
16-
# In NVIDIA's image, there's no CUDA_HOME or CUDA_PATH.
17-
# Also, the LD_LIBRARY_PATH is intended to point to a non-existent folder.
14+
# Note about environment variables in NVIDIA's image:
15+
# 1. There's no CUDA_HOME or CUDA_PATH.
16+
# 2. The LD_LIBRARY_PATH was intended to point to a non-existent folder.
1817
# Ref: https://gitlab.com/nvidia/container-images/cuda/-/issues/47
1918
################################################################################
2019

@@ -53,6 +52,8 @@ vim \
5352
ca-certificates \
5453
libjpeg-dev \
5554
libpng-dev \
55+
# This meta package will install libcusparselt0-cuda-13 & libcusparselt0-dev-cuda-13
56+
cusparselt-cuda-13 \
5657
# ffmpeg \
5758
# x264 \
5859
# x265 \
@@ -74,9 +75,10 @@ RUN --mount=type=cache,target=/root/.cache/pip \
7475
################################################################################
7576
# PyTorch
7677
# To reduce image size & single layer size, we:
77-
# 1. Install PyTorch only.
78-
# 2. Install all its dependencies.
79-
# 3. Delete redundant NVIDIA Python libs (they are already installed as OS packages).
78+
# 1. Fetch the list of packages and download them into the cache.
79+
# 2. Install PyTorch only.
80+
# 3. Install all dependencies.
81+
# 4. Delete redundant NVIDIA Python libs (they are already installed as OS packages).
8082
RUN --mount=type=cache,target=/root/.cache/pip \
8183
pip install \
8284
--dry-run --pre torch torchvision torchaudio \
@@ -129,4 +131,4 @@ ENV PATH="${PATH}:/root/.local/bin"
129131
# Suppress [WARNING: Running pip as the 'root' user]
130132
ENV PIP_ROOT_USER_ACTION=ignore
131133

132-
CMD ["bash", "-c", "python3.14t ./ComfyUI/main.py --listen --port 8188 $CLI_ARGS"]
134+
CMD ["bash", "-c", "python3.14t ./ComfyUI/main.py --listen --port 8188 ${CLI_ARGS}"]

0 commit comments

Comments
 (0)