pip安装Pytorch GPU版

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126

也可以到https://download.pytorch.org/whl/torch/网站，下载whl

然后：pip install torch-2.6.0+cu126-cp312-cp312-win_amd64.whl