logo

Anaconda 中使用 Conda 和 Pip 安装 PyTorch CPU 版本

作者:公子世无双2024.01.17 23:03浏览量:248

简介:本文介绍了在 Anaconda 中使用 Conda 和 Pip 安装 PyTorch CPU 版本的详细步骤,包括创建虚拟环境、安装对应版本的 PyTorch 等。亲测有效,适合初学者和进阶用户。

在 Anaconda 中使用 Conda 和 Pip 安装 PyTorch CPU 版本是一项相对简单的任务。以下是详细步骤:
方法一:使用 Conda

  1. 创建虚拟环境:首先,你需要创建一个虚拟环境。在 Anaconda Prompt 中,输入以下命令来创建一个名为 pytorch 的虚拟环境,并指定 Python 版本为 3.8。
    1. conda create --name pytorch python=3.8
  2. 激活虚拟环境:创建完成后,你需要激活这个环境。
    1. conda activate pytorch
  3. 安装 PyTorch:在激活的虚拟环境中,使用以下命令安装 PyTorch 的 CPU 版本。这将从 PyTorch 的官方 Conda 仓库中下载并安装最新版本的 PyTorch。
    1. conda install pytorch torchvision torchaudio cpuonly -c pytorch
  4. 验证安装:安装完成后,你可以通过运行以下代码来验证 PyTorch 是否正确安装。
    1. import torch
    2. print(torch.__version__)
    如果成功输出了 PyTorch 的版本号,则表示安装成功。
    方法二:使用 Pip
    如果你更喜欢使用 Pip 来安装 PyTorch,也可以按照以下步骤操作:
  5. 创建虚拟环境:同样,首先创建一个虚拟环境。在 Anaconda Prompt 中,输入以下命令创建一个名为 pytorch 的虚拟环境,并指定 Python 版本为 3.8。
    1. conda create -n pytorch python=3.8
  6. 激活虚拟环境:创建完成后,激活这个环境。
    1. conda activate pytorch
  7. 安装 PyTorch:在激活的虚拟环境中,使用以下命令来安装 PyTorch 的 CPU 版本。这将从 PyTorch 的官网下载并安装最新版本的 PyTorch。注意,你可能需要先下载对应版本的预编译包,或者耐心等待自动下载和安装完成。
    1. pip install torch torchvision torchaudio cpuonly
  8. 验证安装:安装完成后,你可以通过运行以下代码来验证 PyTorch 是否正确安装。
    1. import torch
    2. print(torch.__version__)
    如果成功输出了 PyTorch 的版本号,则表示安装成功。
    总结:在 Anaconda 中使用 Conda 或 Pip 安装 PyTorch CPU 版本都是可行的选择。使用 Conda 可以方便地管理依赖关系和虚拟环境,而使用 Pip 则更加灵活和便捷。无论你选择哪种方式,都请确保按照正确的步骤进行操作,并遵循最佳实践来管理虚拟环境和依赖关系。在安装过程中遇到问题时,可以查阅相关文档或寻求社区帮助。

相关文章推荐

发表评论

活动