Skip to content

Commit 99594a5

Browse files
committed
adding libcufile.so.1.14.0 to the cpu only container so that libtorch models can load
1 parent 30e20cb commit 99594a5

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

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.1.14.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)