diff --git a/qa/TL2_FW_iterators_perf/test_pytorch.sh b/qa/TL2_FW_iterators_perf/test_pytorch.sh index 5df1a4a74e3..0d7df655f9a 100755 --- a/qa/TL2_FW_iterators_perf/test_pytorch.sh +++ b/qa/TL2_FW_iterators_perf/test_pytorch.sh @@ -9,6 +9,9 @@ do_once() { } test_body() { + # turn off SHARP to avoid NCCL errors + export NCCL_NVLS_ENABLE=0 + for fw in "pytorch"; do python test_RN50_data_fw_iterators.py --framework ${fw} --gpus ${NUM_GPUS} -b 13 \ --workers 3 --prefetch 2 --epochs 3