PyTorch GPU Edition

CUDA has an extension for Visual Studio, so firstly, install VS IDE,

then,

https://developer.nvidia.com/cuda-11-8-0-download-archive
https://developer.nvidia.com/rdp/cudnn-download
https://download.pytorch.org/whl/cu118/torch-2.0.1%2Bcu118-cp311-cp311-win_amd64.whl
https://download.pytorch.org/whl/cu118/torchvision-0.15.2%2Bcu118-cp311-cp311-win_amd64.whl

refer to:
https://zhuanlan.zhihu.com/p/305854081