File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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# ---------------------------------------------------------------------------
9194RUN 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
You can’t perform that action at this time.
0 commit comments