pytorch安装 gpu cpu

要安装 PyTorch，需要根据自己的环境选择相应的安装方式。如果你的电脑有 NVIDIA 显卡并且已经安装了 CUDA，那么可以选择安装 GPU 版本的 PyTorch，否则可以选择安装 CPU 版本的 PyTorch。

安装 GPU 版本的 PyTorch，可以使用以下命令：

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

其中，`cu111` 表示 CUDA 11.1 版本，如果你的 CUDA 版本不同，需要修改为相应的版本号。

安装 CPU 版本的 PyTorch，可以使用以下命令：

pip install torch torchvision torchaudio

这样就可以安装 PyTorch 了。