Skip to content

Commit 494a6af

Browse files
committed
Removed the nccl req.
1 parent 050bd92 commit 494a6af

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

‎.devcontainer/runtime.Dockerfile‎

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -87,6 +87,9 @@ RUN dnf upgrade --refresh -y && \
8787
# CUDA RUNTIME libraries only (NOT the toolkit) via NVIDIA's network repo -- a few
8888
# hundred MB of .so instead of the ~4-5GB local toolkit installer. These are exactly the
8989
# libs the prebuilt editable torch links against.
90+
# NOTE: NCCL is NOT installed here -- it isn't in NVIDIA's Fedora CUDA repo (that repo
91+
# only ships nccl on Debian/Ubuntu). torch bundles its own via the nvidia-nccl-cu12
92+
# wheel, which lives in the PVC's site-packages and is found at runtime.
9093
# ---------------------------------------------------------------------------
9194
RUN CUDA_MAJMIN="$(echo "${CUDA_VERSION}" | tr . -)" && \
9295
CUDA_DIR="cuda-${CUDA_VERSION}" && \
@@ -102,8 +105,7 @@ RUN CUDA_MAJMIN="$(echo "${CUDA_VERSION}" | tr . -)" && \
102105
libcusparse-${CUDA_MAJMIN} \
103106
libnpp-${CUDA_MAJMIN} \
104107
libnvjitlink-${CUDA_MAJMIN} \
105-
libcufile-${CUDA_MAJMIN} \
106-
libnccl && \
108+
libcufile-${CUDA_MAJMIN} && \
107109
dnf clean all && \
108110
echo /usr/local/${CUDA_DIR}/lib64 > /etc/ld.so.conf.d/cuda-runtime.conf
109111

0 commit comments

Comments
 (0)