If you have a conda environment, create a new environement using
conda create --name myenv
activate myenv
Install pytorch in the environment. if you have cuda: check commandline corresponding to your version.
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch