search

NVIDIA-Merlin / Transformers4Rec

Transformers4Rec is a flexible and efficient library for sequential and session-based recommendation and works with PyTorch.
https://nvidia-merlin.github.io/Transformers4Rec/main
Apache License 2.0
1.08k stars 142 forks source link

Add pytorch backend path to LD_LIBRARY_PATH in integration test env #715

Closed oliverholworthy closed 1 year ago

oliverholworthy commented 1 year ago

Adds pytorch backend path to LD_LIBRARY_PATH in integration test env

Fixes Error about undefined symbol loading mkl. When running Triton with a pytorch model.

INTEL MKL ERROR: /usr/local/lib/libmkl_vml_avx2.so.1: undefined symbol: mkl_lapack_dspevd.
Intel MKL FATAL ERROR: cannot load libmkl_vml_avx2.so.1 or libmkl_vml_def.so.1.