pytorch怎么安装

pengjiani TOP1 725

pytorch安装需要:1.查看安装环境;2.安装Anaconda和pip。pytorch 可以在各种 Windows 发行版上安装和使用。根据系统和计算要求,Windows 上使用 pytorch 的体验可能会在处理时间方面有所不同。

pytorch怎么安装

1.查看安装环境

1)支持的Windows版本

以下 Windows 发行版支持 PyTorch:

  • Windows 7 及更高版本;建议使用Windows 10或更高版本。
  • Windows Server 2008 r2 及更高版本

2)python

目前,Windows 上的 PyTorch 仅支持 Python 3.7-3.9;不支持 Python 2.x。

3)包管理器

要安装 PyTorch 二进制文件,您至少需要使用两个受支持的包管理器之一:Anaconda和pip。Anaconda 是推荐的包管理器,因为它将在一个沙盒安装中为您提供所有 PyTorch 依赖项,包括 Python 和pip.

2.安装

1)Anaconda

要使用 Anaconda 安装 PyTorch,您需要通过Start | Anaconda3 | Anaconda Prompt.

  • 没有 CUDA

要通过 Anaconda 安装 PyTorch,并且没有支持 CUDA 的系统或不需要 CUDA,请在上述选择器中选择 OS:Windows、Package:Conda 和 CUDA:None。然后,运行显示给您的命令。

  • 有CUDA

要通过 Anaconda 安装 PyTorch,并且您确实拥有支持 CUDA 的系统,请在上述选择器中选择 OS:Windows、Package:Conda 和适合您机器的 CUDA 版本。通常,最新的 CUDA 版本会更好。然后,运行显示给您的命令。

2)pip

  • 没有CUDA

要通过 pip 安装 PyTorch,并且没有支持 CUDA 的系统或不需要 CUDA,请在上述选择器中选择 OS:Windows、Package:Pip 和 CUDA:None。然后,运行显示给您的命令。

  • 有CUDA

要通过 pip 安装 PyTorch,并拥有支持 CUDA 的系统,请在上述选择器中选择 OS:Windows、Package:Pip 和适合您机器的 CUDA 版本。通常,最新的 CUDA 版本会更好。然后,运行显示给您的命令。

3.确认安装

为了确保 PyTorch 安装正确,我们可以通过运行示例 PyTorch 代码来验证安装。在这里,我们将构造一个随机初始化的张量。

在命令行中,键入:

python

然后输入以下代码:

import torch
x = torch.rand(5, 3)
print(x)

输出应该类似于:

tensor([[0.3380, 0.3845, 0.3217],
        [0.8337, 0.9050, 0.2650],
        [0.2979, 0.7141, 0.9069],
        [0.1449, 0.1132, 0.1375],
        [0.4675, 0.3947, 0.1426]])

此外,要检查您的 GPU 驱动程序和 CUDA 是否已启用并可由 PyTorch 访问,请运行以下命令以返回 CUDA 驱动程序是否已启用:

import torch
torch.cuda.is_available()


最后,推荐我们的管理工具给大家。

回复

我来回复
  • 暂无回复内容

注册PingCode 在线客服
站长微信
站长微信
电话联系

400-800-1024

工作日9:30-21:00在线

分享本页
返回顶部