Pytorch 在Anaconda中使用conda或pip安装Pytorch的方法

在本文中，我们将介绍如何在Anaconda中使用conda或pip安装Pytorch，并提供示例说明。

使用conda安装Pytorch

conda是一个开源的包管理器，可以用于安装、更新、卸载软件包。使用conda安装Pytorch十分方便。下面是安装Pytorch的步骤：

打开命令行（Windows用户为Anaconda Prompt，Mac和Linux用户为终端）。
创建一个新的conda环境（可选）。若要创建一个新的环境，请运行以下命令：

conda create -n pytorch_env

其中，pytorch_env是环境的名称，您可以根据自己的需求进行修改。

激活环境。若已创建了新的环境，请运行以下命令：

conda activate pytorch_env

使用conda安装Pytorch。运行以下命令：

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge

安装完成后，可以通过以下命令检查Pytorch是否安装成功：

python -c "import torch; print(torch.__version__)"

如果成功安装，将会显示Pytorch版本号。

使用pip安装Pytorch

pip是Python的包管理工具，使用pip安装Pytorch同样也非常方便。下面是安装Pytorch的步骤：

打开命令行（Windows用户为命令提示符，Mac和Linux用户为终端）。
创建一个新的虚拟环境（可选）。若要创建一个新的环境，请运行以下命令：

python -m venv pytorch_env

其中，pytorch_env是环境的名称，您可以根据自己的需求进行修改。

激活环境。若已创建了新的虚拟环境，请运行以下命令：

对于Windows用户：

.\pytorch_env\Scripts\activate

对于Mac和Linux用户：

source pytorch_env/bin/activate

使用pip安装Pytorch。运行以下命令：

pip install torch torchvision torchaudio

安装完成后，可以通过以下命令检查Pytorch是否安装成功：

python -c "import torch; print(torch.__version__)"

如果成功安装，将会显示Pytorch版本号。

示例说明

下面是一些示例说明，演示了在Anaconda中使用conda或pip安装Pytorch的具体过程。

示例1：使用conda安装Pytorch

conda create -n pytorch_env
conda activate pytorch_env
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge
python -c "import torch; print(torch.__version__)"

示例2：使用pip安装Pytorch

python -m venv pytorch_env
source pytorch_env/bin/activate
pip install torch torchvision torchaudio
python -c "import torch; print(torch.__version__)"

通过以上示例，您可以快速了解在Anaconda中使用conda或pip安装Pytorch的方法。