TAO Toolkit deep learning networks with PyTorch backend
conda install conda-forge::nvidia-tao-pytorch