如何在Win10上安装PyTorch

目录

准备工作

在开始安装PyTorch之前,我们需要确保计算机上已经安装了一些必要的软件和库。请按照以下步骤进行准备:

  1. 确保你的计算机上已经安装了Windows 10操作系统。
  2. 确保你的计算机上安装了Python。你可以在命令行中输入python --version来检查Python的版本。如果没有安装Python,请前往[Python官方网站](
  3. 确保你的计算机上安装了pip。你可以在命令行中输入pip --version来检查pip的版本。如果没有安装pip,请按照以下步骤安装pip:
    • 下载[pip安装脚本](
    • 在命令行中切换到下载的脚本所在的目录
    • 运行命令python get-pip.py进行安装

完成了以上准备工作后,我们可以开始安装PyTorch了。

安装Anaconda

在开始安装PyTorch之前,我们先安装Anaconda,它是一个Python的科学计算发行版,集成了常用的科学计算库和工具。

  1. 前往[Anaconda官方网站](
  2. 双击安装程序并按照提示完成安装。在安装过程中,需要选择是否将Anaconda添加到系统的环境变量中,请确保选择了这个选项。
  3. 安装完成后,打开Anaconda Navigator。你可以在Windows的开始菜单中找到它。

创建虚拟环境

在安装PyTorch之前,我们可以创建一个独立的Python虚拟环境,以便隔离不同项目所使用的库和工具。

  1. 在Anaconda Navigator中,点击左侧导航栏中的“环境”选项。
  2. 点击右上角的“+ Create”按钮,弹出一个新建环境的对话框。
  3. 输入一个环境名称,比如“pytorch_env”。
  4. 在“Packages”列表中选择“Python”和“pip”这两个包,然后点击“Apply”按钮,开始创建环境。

创建虚拟环境需要一些时间,等待过程完成后,我们可以安装PyTorch了。

安装PyTorch

在这一步中,我们将使用pip安装PyTorch。

  1. 在Anaconda Navigator中,确保已经选择了之前创建的虚拟环境“pytorch_env”。

  2. 在右侧的“Home”选项中,选择“pip install”。

  3. 在弹出的对话框中,输入以下命令并点击“Install”按钮:

    pytorch
    

    这个命令将会使用pip安装最新版本的PyTorch。

等待安装完成后,我们可以验证一下是否安装成功。

验证安装

我们可以编写一个简单的Python脚本来验证PyTorch的安装情况。

  1. 打开任意文本编辑器,创建一个新的Python脚本文件,比如pytorch_test.py

  2. 在文件中输入以下代码:

    import torch
    
    x = torch.tensor([1.0, 2.0, 3.0, 4.0])
    y = torch.tensor([5.0, 6.0, 7.0, 8.0])
    
    z = x + y
    
    print(z)
    

    这段代码导入了PyTorch库并创建了两个张量(tensor),然后将它们相加并打印结果。

  3. 保存文件,并使用命