From 755439d8749c8558e7896a34718f47cbf125fad0 Mon Sep 17 00:00:00 2001 From: Joaquin Anton Date: Fri, 14 Jun 2024 09:35:27 +0100 Subject: [PATCH] Fix TL3_RN50_short/test_mxnet.sh to load nvimagecodec first to avoid libjpeg symbol clash (#5522) Signed-off-by: Joaquin Anton --- qa/TL3_RN50_short/test_mxnet.sh | 3 +++ 1 file changed, 3 insertions(+) diff --git a/qa/TL3_RN50_short/test_mxnet.sh b/qa/TL3_RN50_short/test_mxnet.sh index 78dc9ac7d0..20b98ddc04 100644 --- a/qa/TL3_RN50_short/test_mxnet.sh +++ b/qa/TL3_RN50_short/test_mxnet.sh @@ -10,6 +10,9 @@ sed -i -e "s/'--dali-nvjpeg-memory-padding', int, 16/'--dali-nvjpeg-memory-paddi # Make sure that nvImageCodec loads before MXNet to avoid getting libjpeg version mismatch sed -i -e "s/import nvidia.dali.fn as fn/import nvidia.dali.fn as fn\nfrom nvidia import nvimgcodec/g" /opt/mxnet/python/mxnet/io/dali_utils.py +# TODO(janton): Remove the LD_PRELOAD with next nvImageCodec release +# Make sure that nvImageCodec's jpeg symbol load before MXNet to avoid getting libjpeg version mismatch +LD_PRELOAD=$(ls /usr/local/lib/python3.10/dist-packages/nvidia/nvimgcodec/extensions/libjpeg_turbo_ext.so*) \ python /opt/mxnet/example/image-classification/train_imagenet_runner \ --data-root=/data/imagenet/train-val-recordio-passthrough/ -b 408 \ -n $NUM_GPUS -e 5 --seed 42 --dali-threads 8 2>&1 | tee dali.log