2023最新pytorch安装(超详细版)
在安装PyTorch之前,请确保您的系统已安装Python和pip。以下是在不同操作系统中安装PyTorch的步骤:
对于Windows
- 打开命令提示符或PowerShell。
- 访问PyTorch官方网站的安装指南:https://pytorch.org/get-started/locally/
- 根据您的系统和需要的PyTorch版本(是否需要CUDA支持),官方网站会提供一个安装命令。
- 在命令提示符或PowerShell中执行该命令。
例如,如果您想要安装不带CUDA支持的PyTorch,可以使用以下命令:
pip install torch torchvision torchaudio
如果需要特定版本的CUDA支持,可以使用类似以下的命令(以CUDA 11.3为例):
pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu113/torch_stable.html
对于macOS
- 打开终端。
- 访问PyTorch官方网站的安装指南:https://pytorch.org/get-started/locally/
- 根据您的系统和需要的PyTorch版本(是否需要CUDA支持),官方网站会提供一个安装命令。
- 在终端中执行该命令。
例如,如果您想要安装不带CUDA支持的PyTorch,可以使用以下命令:
pip install torch torchvision torchaudio
如果需要特定版本的CUDA支持,可以使用类似以下的命令(以CUDA 11.1为例):
pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu113/torch_stable.html
对于Linux
对于Linux,您可以使用pip或conda来安装PyTorch。
使用pip安装(不带CUDA支持):
pip install torch torchvision torchaudio
使用conda安装(不带CUDA支持):
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
请根据您的CUDA版本调整cudatoolkit
的版本。
注意:在安装之前,请确保您的系统已经安装了CUDA(如果需要CUDA支持的话)。
评论已关闭