PyTorch 在 Windows 10 系统下的环境配置及安装

本文详述了在 Windows 10 系统中配置 PyTorch 的步骤,包括先安装 Anaconda,检查或安装 NVIDIA 显卡驱动和 CUDA,下载 cuDNN,以及在 Anaconda 环境中创建和激活 PyTorch 环境,最终安装 PyTorch。此外,还提供了验证安装成功的测试代码。
摘要由CSDN通过智能技术生成

PyTorch 简介

PyTorch 是一个开源的 Python 机器学习库,基于 Torch,用于自然语言处理等应用程序。

相关链接:

本文初次撰写于 2020-09-28,最后更新于 2020-09-29.


安装 PyTorch 之前


安装 Anaconda

Anaconda 是一个集成了大量用于科学分析(包括机器学习,深度学习等)的 package 的工具包,可以帮助我们管理相关的 package,是在单台机器上执行机器学习的最简单方法。

本文采用 Anaconda 作为 PyTorch 包的管理程序,您也可以选择 pip 等工具进行管理,不再赘述。

您可以点击此链接前往 Anaconda 的官方网站,下载并安装适用于个人用户学习使用的开源版程序。

Download Anaconda
安装成功后,您可以通过 Windows 10 的搜索功能或开始菜单轻松找到 Anaconda Powershell Prompt 或 Anaconda Prompt. 两者打开方式本质上没有区别,您可以任意选择。本文的后续演示将使用 Anaconda Powershell Prompt 进行操作。

Search Anaconda Prompt


假如您有 NVIDIA 系列显卡

假如没有,您可以跳过这一步,后续 PyTorch 的安装过程不会受到影响。

检验驱动程序

显卡可以大幅度提升机器学习时的处理速度。打开任务管理器 - 性能 - GPU, 假如您已经安装显卡驱动,那么会在此处(图中蓝框勾出)显示具体的显卡型号。假如您确实有 NVIDIA 显卡但在此处没有显示,可以前往 NVIDIA 官网下载对应的驱动程序并安装。

GPU

安装 CUDA

CUDA 是用于GPU编程的语言ÿ

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值