PyTorch is an optimized tensor library for deep learning using GPUs and CPUs.
conda install esri-build::pytorch-gpu
conda install esri-build/label/cuda-12::pytorch-gpu