NK990

Changed cuda:1 to cuda:0

Installing this: `pip install torch==1.12.0+cu116 torchvision==0.13.0+cu116 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu116` issue may be a solution!