步骤一:安装Anaconda

首先,我们需要先安装Anaconda。Anaconda是一个Python数据科学平台,提供了包管理和环境管理的功能。安装Anaconda可以简化PyTorch的安装过程。你可以在Anaconda官网上下载对应操作系统的安装程序,然后按照提示进行安装。

步骤二:创建虚拟环境

为了隔离不同项目所使用的不同Python包,我们需要创建一个虚拟环境。在命令行中,输入以下命令创建一个名为"pytorch"的虚拟环境:

conda create -n pytorch python=3.7

这将创建一个名为"pytorch"的虚拟环境,并且使用Python 3.7版本。你可以根据你的需要选择其他Python版本。

步骤三:激活虚拟环境并安装PyTorch

当虚拟环境创建完成后,我们需要激活该环境并安装PyTorch。在命令行中,输入以下命令来激活虚拟环境:

conda activate pytorch

激活虚拟环境后,我们可以使用conda来安装PyTorch。输入以下命令来安装PyTorch及其相关依赖:

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

这将安装PyTorch及其相关的依赖包。如果你的电脑上有NVIDIA的显卡并安装了CUDA,那么你可以使用GPU来加速PyTorch的计算。需要注意的是,你需要根据你的CUDA版本来选择正确的cudatoolkit版本。