Skip to content

Commit

Permalink
Fix TL3_RN50_short/test_mxnet.sh to load nvimagecodec first to avoid …
Browse files Browse the repository at this point in the history
…libjpeg symbol clash (#5522)

Signed-off-by: Joaquin Anton <janton@nvidia.com>
  • Loading branch information
jantonguirao committed Jun 14, 2024
1 parent 927c93b commit 755439d
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions qa/TL3_RN50_short/test_mxnet.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 755439d

Please sign in to comment.