在 Windows 上安装 PyTorch GPU

PyTorch 是一个适用于 Python 的科学计算框架，它可以用于创建深度学习模型。如果你想在 Windows 上使用 PyTorch GPU 版本，那么本文将为你提供详细的安装步骤。

步骤

1. 安装 Anaconda

Anaconda 是一个流行的 Python 发行版，它包含了许多流行的科学计算和机器学习库。我们可以从官方网站下载安装程序并安装 Anaconda。

2. 创建虚拟环境

为了避免版本冲突和其他问题，我们建议使用 Anaconda 创建一个新的虚拟环境。可以使用以下命令创建名为“pytorch”的虚拟环境。

conda create --name pytorch python=3.7

3. 激活虚拟环境

我们创建完新的虚拟环境之后，需要使用以下命令激活虚拟环境。

conda activate pytorch

4. 安装 PyTorch GPU 版本

在激活的虚拟环境下，我们可以使用以下命令安装 PyTorch GPU 版本。

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

此命令将安装最新版本的 PyTorch GPU 版本并包含 TorchVision 和 TorchAudio 库，同时与 CUDA 11.1 兼容。

5. 验证安装

安装完成后，我们可以使用以下代码验证 PyTorch 是否成功安装。

import torch

if torch.cuda.is_available():
    print('PyTorch GPU 版本已安装成功！')
else:
    print('PyTorch GPU 版本安装失败！')

如果你在你的控制台看到了“PyTorch GPU 版本已安装成功！”消息，则说明你已成功安装 PyTorch GPU 版本。

结论

在本文中，我们详细介绍了如何在 Windows 上安装 PyTorch GPU 版本。请按照上述步骤操作，确保正确安装。