cuda12.1装gpu版本的pytorch的pip命令

在安装 CUDA 12.1 的情况下，安装 GPU 版本的 PyTorch 可以使用以下命令：

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

该命令会安装 PyTorch 1.8.1、torchvision 0.9.1 和 torchaudio 0.8.1，适用于 CUDA 11.1。如果需要安装其他版本的 PyTorch，可以根据需要更改版本号。