pytorch怎么安装
pytorch安装需要:1.查看安装环境;2.安装Anaconda和pip。pytorch 可以在各种 Windows 发行版上安装和使用。根据系统和计算要求,Windows 上使用 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()
最后,推荐我们的管理工具给大家。