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