简易入门安装PyTorch及其测试验证

目录

检查系统要求

安装PyTorch 

通过pip在windows环境中安装GPU版本:

使用conda方式在windows中安装GPU版本

 在windows环境安装cpu版本

验证安装 

总结


检查系统要求

在开始安装PyTorch之前,请确保你的系统满足以下要求:

  • Python版本:3.6或更高
  • 操作系统:Windows,macOS,或Linux

安装PyTorch 

        安装pytorch官网直接给出了安装的命令行【PyTorch】:

        在这里我们可以通过选择不同版本,不同安装方式和选择不同的硬件去进行安装,比如:

通过pip在windows环境中安装GPU版本:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

这段代码是用于安装PyTorch及其相关库的Python命令。下面是对这个命令的详细解释:

  • pip3: 这是Python的包管理工具pip的一个版本,专门用于Python 3。pip允许你安装和管理Python包。

  • install: 这是pip命令中用于安装包的关键字。

  • torch: 这是PyTorch的主要包,一个用于机器学习和深度学习的开源库。它提供了强大的张量计算(类似于NumPy数组)和自动微分功能,非常适合进行机器学习和深度学习的研究和开发。

  • torchvision: 这是与PyTorch配合使用的一个包,专注于图像处理。它提供了常用的数据集、模型架构和图像转换工具,非常适合于计算机视觉任务。

  • torchaudio: 这是另一个与PyTorch配合使用的包,专注于音频处理。它提供了音频处理所需的数据集、模型和转换工具。

  • --index-url: 这个参数指定了一个替代的Python包索引,pip会从这个索引中搜索和下载包。

  • https://download.pytorch.org/whl/cu121: 这是上面--index-url参数指定的URL,它是PyTorch官方提供的一个包含预编译轮(wheel)文件的服务器。这些轮文件是针对特定版本的CUDA(cu121代表CUDA 12.1)预编译的,这意味着如果你的系统安装了相应版本的CUDA,PyTorch可以利用GPU进行加速。

使用conda方式在win

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

E寻数据

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值