指定安装pytorch gpu版本

要安装PyTorch GPU版本，请使用以下命令：

pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html

此命令将安装适用于CUDA 11.1的PyTorch GPU版本。如果您的CUDA版本不同，您可以在链接中更改版本号。