Win10使用清华源快速安装PyTorch-GPU版(推荐)

在Win10上安装PyTorch-GPU版可以加速深度学习模型的训练。本文将介绍如何使用清华源快速安装PyTorch-GPU版，并提供两个示例。

安装Anaconda

首先，我们需要安装Anaconda，它是一个流行的Python发行版，包含了许多常用的Python库和工具。您可以从官方网站下载适用于您的操作系统的Anaconda安装程序，并按照提示进行安装。

创建虚拟环境

接下来，我们需要创建一个虚拟环境，以便在其中安装PyTorch-GPU版。在Anaconda Prompt中输入以下命令：

conda create --name pytorch_gpu python=3.8

这将创建一个名为pytorch_gpu的虚拟环境，并使用Python 3.8版本。

安装PyTorch-GPU版

接下来，我们需要使用清华源安装PyTorch-GPU版。在Anaconda Prompt中输入以下命令：

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

这将使用清华源安装PyTorch-GPU版及其依赖项。请注意，cudatoolkit参数的值应与您的CUDA版本匹配。

测试PyTorch-GPU版

安装完成后，我们可以使用以下示例测试PyTorch-GPU版是否正确安装：

import torch

if torch.cuda.is_available():
    device = torch.device("cuda")
    x = torch.randn(1000, 1000, device=device)
    y = torch.randn(1000, 1000, device=device)
    z = torch.matmul(x, y)
    print(z)
else:
    print("CUDA is not available")

如果输出了一个1000x1000的矩阵，则说明PyTorch-GPU版已经正确安装。

总结

在本文中，我们介绍了如何使用清华源快速安装PyTorch-GPU版，并提供了一个测试示例。如果您按照这些步骤进行操作，您应该能够在Win10上成功安装PyTorch-GPU版。