Установите графический процессор Pytorch на Windows

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
Pleasant Pheasant