使用pip安装cuda11.2对应的pytorch-gpu

您可以通过以下命令使用pip安装与CUDA 11.2兼容的pytorch-gpu：

pip install torch==1.8.1+cu112 torchvision==0.9.1+cu112 torchaudio==0.8.1 -f https://download.pytorch.org/whl/cu112/torch_stable.html

这将安装pytorch版本为1.8.1，同时与CUDA 11.2兼容。请确保您的系统已正确安装了CUDA 11.2，并且您的环境已正确配置。