Cannot train Tao Toolkit UNet model in version v4.0.0 and v4.0.1 - TAO Toolkit - NVIDIA Developer Forums

Excuse me @Bin_Zhao_NV @Morganh I’ve changed gpus from Tesla P100 to Tesla V100 and tried to train Tao Toolkit UNet model with 4 gpus in version v4.0.0 and v4.0.1 again. However. I still got the error message: device CUDA:0 not supported by XLA service while setting up XLA_GPU_JIT device number 0. This is the result in the process of training UNet when I ran the command nvidia-smi. Is this a bug for Tao Toolkit v4.0.0 and v4.0.1 ? When I trained UNet in the version v3.22.05, it seemed tha

TypeError: get_calibrator() got an unexpected keyword argument 'image_mean' - TAO Toolkit - NVIDIA Developer Forums

The training process of Tao-Toolkit-API unet is always in Inf status - TAO Toolkit - NVIDIA Developer Forums

Smart Pruning: Improve mobile ML performance - AI and ML blog - Arm Community blogs - Arm Community

NVIDIA TAO Toolkit 'Zero to Hero': Setup Tips & Tricks

The training process of Tao-Toolkit-API unet is always in Inf status - TAO Toolkit - NVIDIA Developer Forums

Cannot train Tao Toolkit UNet model in version v4.0.0 and v4.0.1 - TAO Toolkit - NVIDIA Developer Forums

Accelerating AI Development with NVIDIA TAO Toolkit and Weights & Biases

Custom TAO unet model classifying only two classes on Deepstream! - TAO Toolkit - NVIDIA Developer Forums

Integrating TAO Models into DeepStream - NVIDIA Docs