Skip to content

Commit 9a88e29

Browse files
committed
build.py - add required installation instructions.
1 parent 6089e81 commit 9a88e29

1 file changed

Lines changed: 10 additions & 0 deletions

File tree

build.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1535,6 +1535,16 @@ def dockerfile_prepare_container_linux(argmap, backends, enable_gpu, target_mach
15351535
repo_arch=repo_arch
15361536
)
15371537

1538+
if target_platform() == "rhel":
1539+
repo_arch = "sbsa" if target_machine == "aarch64" else "x86_64"
1540+
df += """
1541+
RUN dnf config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel8/{repo_arch}/cuda-rhel8.repo \\
1542+
&& dnf clean expire-cache \\
1543+
&& dnf install --assumeyes libnvshmem3-cuda-13
1544+
""".format(
1545+
repo_arch=repo_arch
1546+
)
1547+
15381548
df += """
15391549
WORKDIR /opt/tritonserver
15401550
RUN rm -fr /opt/tritonserver/*

0 commit comments

Comments
 (0)