Update the pytorch version in repository version to compatible with newer GPU

[TranNg65]

Hi hmorimitsu,

I got this warning when using the repository with RTX 3090:

"NVIDIA GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70."

Can you update the codes for it to be compatible with newer GPUs?

Thank you for your good works.

[TranNg65]

the issue is solved after reinstalling pytorch.