Installation problems with pytorch

[KimamanaNeko]

You recommand installing pytorch with pip.
But I found that the default installation is the cpu version of PyTorch.

I tried to test your code and got the error:
AssertionError: Torch not compiled with CUDA enabled

It is recommended to specify the GPU version of PyTorch to be used in the documentation
BTW I used this code to install PyTorch and fix the cuda error reporting problem
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio===0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

[Equim-chan]

This had already been documented, read it carefully.

[...] Check pytorch's doc on how to install pytorch in your environment. Personally, I recommand installing pytorch with pip.