Get the command configuration from here: https://pytorch.org/get-started/locally/

conda install pytorch torchvision -c pytorch