conda install pytorch torchvision torchaudio -c pytorch