From cb90c2ce3ca953c006bf21edd6e2884f1472f00f Mon Sep 17 00:00:00 2001 From: "Wang, Yang Y" Date: Tue, 18 Jun 2024 18:23:59 +0000 Subject: [PATCH] fix habanalabs-thunk build error with prior version of synapseai GS-295 --- dockerfiles/triton/Dockerfile | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/dockerfiles/triton/Dockerfile b/dockerfiles/triton/Dockerfile index 6902eee..897ec46 100644 --- a/dockerfiles/triton/Dockerfile +++ b/dockerfiles/triton/Dockerfile @@ -29,7 +29,8 @@ ENV PYTHONPATH=/root:/usr/lib/habanalabs/ RUN echo "deb https://${ARTIFACTORY_URL}/artifactory/debian jammy main" | tee -a /etc/apt/sources.list && \ wget "https://${ARTIFACTORY_URL}/artifactory/api/gpg/key/public" && \ apt-key add public && rm public && apt-get update && \ - apt-get install -y habanalabs-thunk="$VERSION"-"$REVISION" \ + apt-get install -y habanalabs-rdma-core="$VERSION"-"$REVISION" \ + habanalabs-thunk="$VERSION"-"$REVISION" \ habanalabs-firmware-tools="$VERSION"-"$REVISION" \ habanalabs-graph="$VERSION"-"$REVISION" && \ apt-get autoremove --yes && apt-get clean && rm -rf /var/lib/apt/lists/* && \ @@ -68,4 +69,4 @@ RUN ln -s /usr/bin/python3.10 /usr/bin/python && wget --no-verbose "${PT_ARTIFAC rm -rf /root/habanalabs pytorch_modules-v"${PT_VERSION}"_"${VERSION}"_"${REVISION}".tgz /tmp/* ENV LD_PRELOAD=/lib/x86_64-linux-gnu/libtcmalloc.so.4 -ENV TCMALLOC_LARGE_ALLOC_REPORT_THRESHOLD=7516192768 \ No newline at end of file +ENV TCMALLOC_LARGE_ALLOC_REPORT_THRESHOLD=7516192768