Skip to content

Commit 9d62236

Browse files
nv-kmcgill53mc-nv
andauthored
fix: adding libcufile.so.0 to the cpu only container so that libtorch models can load (#8265)
Co-authored-by: Misha Chornyi <[email protected]>
1 parent d17512b commit 9d62236

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

build.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1573,6 +1573,7 @@ def add_cpu_libs_to_linux_dockerfile(backends, target_machine):
15731573
COPY --from=min_container /usr/local/cuda/lib64/libcudart.so.12 /usr/local/cuda/targets/{cuda_arch}-linux/lib/.
15741574
COPY --from=min_container /usr/local/cuda/lib64/libcupti.so.12 /usr/local/cuda/targets/{cuda_arch}-linux/lib/.
15751575
COPY --from=min_container /usr/local/cuda/lib64/libnvJitLink.so.12 /usr/local/cuda/targets/{cuda_arch}-linux/lib/.
1576+
COPY --from=min_container /usr/local/cuda/lib64/libcufile.so.0 /usr/local/cuda/targets/{cuda_arch}-linux/lib/.
15761577
15771578
RUN mkdir -p /opt/hpcx/ucc/lib/ /opt/hpcx/ucx/lib/
15781579
COPY --from=min_container /opt/hpcx/ucc/lib/libucc.so.1 /opt/hpcx/ucc/lib/libucc.so.1

0 commit comments

Comments
 (0)