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