PyTorch is an optimized tensor library for deep learning using GPUs and CPUs.
conda install conda-forge::pytorch-gpu