以下是 PyTorch 在不同操作系统上的安装步骤:
Windows
- 打开 PyTorch 官网下载页面:PyTorch Windows 下载
- 选择合适的版本进行下载。
- 运行安装程序,并按照提示完成安装。
macOS
- 打开终端。
- 输入以下命令安装:
pip install torch torchvision torchaudio
- 如果需要 GPU 支持,还需要安装 CUDA:
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
Linux
- 打开终端。
- 输入以下命令安装:
pip install torch torchvision torchaudio
- 如果需要 GPU 支持,还需要安装 CUDA:
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
注意事项
- 确保您的 Python 版本与 PyTorch 兼容。
- 如果遇到安装问题,可以参考 PyTorch 安装常见问题解答
[center]
PyTorch Logo