NOTE: Python 3.9 users will need to add '-c=conda-forge' for installationconda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch