如何在Win10上安装PyTorch
目录
准备工作
在开始安装PyTorch之前,我们需要确保计算机上已经安装了一些必要的软件和库。请按照以下步骤进行准备:
- 确保你的计算机上已经安装了Windows 10操作系统。
- 确保你的计算机上安装了Python。你可以在命令行中输入
python --version
来检查Python的版本。如果没有安装Python,请前往[Python官方网站]( - 确保你的计算机上安装了pip。你可以在命令行中输入
pip --version
来检查pip的版本。如果没有安装pip,请按照以下步骤安装pip:- 下载[pip安装脚本](
- 在命令行中切换到下载的脚本所在的目录
- 运行命令
python get-pip.py
进行安装
完成了以上准备工作后,我们可以开始安装PyTorch了。
安装Anaconda
在开始安装PyTorch之前,我们先安装Anaconda,它是一个Python的科学计算发行版,集成了常用的科学计算库和工具。
- 前往[Anaconda官方网站](
- 双击安装程序并按照提示完成安装。在安装过程中,需要选择是否将Anaconda添加到系统的环境变量中,请确保选择了这个选项。
- 安装完成后,打开Anaconda Navigator。你可以在Windows的开始菜单中找到它。
创建虚拟环境
在安装PyTorch之前,我们可以创建一个独立的Python虚拟环境,以便隔离不同项目所使用的库和工具。
- 在Anaconda Navigator中,点击左侧导航栏中的“环境”选项。
- 点击右上角的“+ Create”按钮,弹出一个新建环境的对话框。
- 输入一个环境名称,比如“pytorch_env”。
- 在“Packages”列表中选择“Python”和“pip”这两个包,然后点击“Apply”按钮,开始创建环境。
创建虚拟环境需要一些时间,等待过程完成后,我们可以安装PyTorch了。
安装PyTorch
在这一步中,我们将使用pip安装PyTorch。
-
在Anaconda Navigator中,确保已经选择了之前创建的虚拟环境“pytorch_env”。
-
在右侧的“Home”选项中,选择“pip install”。
-
在弹出的对话框中,输入以下命令并点击“Install”按钮:
pytorch
这个命令将会使用pip安装最新版本的PyTorch。
等待安装完成后,我们可以验证一下是否安装成功。
验证安装
我们可以编写一个简单的Python脚本来验证PyTorch的安装情况。
-
打开任意文本编辑器,创建一个新的Python脚本文件,比如
pytorch_test.py
。 -
在文件中输入以下代码:
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),然后将它们相加并打印结果。
-
保存文件,并使用命