以下是 PyTorch 在不同操作系统上的安装步骤:

Windows

  1. 打开 PyTorch 官网下载页面:PyTorch Windows 下载
  2. 选择合适的版本进行下载。
  3. 运行安装程序,并按照提示完成安装。

macOS

  1. 打开终端。
  2. 输入以下命令安装:pip install torch torchvision torchaudio
  3. 如果需要 GPU 支持,还需要安装 CUDA:conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

Linux

  1. 打开终端。
  2. 输入以下命令安装:pip install torch torchvision torchaudio
  3. 如果需要 GPU 支持,还需要安装 CUDA:conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

注意事项

[center]

PyTorch Logo
[center]