Fix LD_LIBRARY_PATH for PyTorch backend

dsgibbons · Jun 30, 2023 · e47fbca · e47fbca
1 parent fff1595
commit e47fbca
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/Dockerfile.QA b/Dockerfile.QA
@@ -371,3 +371,6 @@ ENV LD_LIBRARY_PATH /opt/tritonserver/qa/clients:${LD_LIBRARY_PATH}
 
 # DLIS-3631: Needed to run Perf Analyzer CI tests correctly
 ENV LD_LIBRARY_PATH /opt/hpcx/ompi/lib:${LD_LIBRARY_PATH}
+
+# Required for PyTorch to pickup the correct HPCX libraries
+ENV LD_LIBRARY_PATH /opt/hpcx/ucc/lib/:/opt/hpcx/ucx/lib/:${LD_LIBRARY_PATH}