PyTorch官方包简介
官方包
| 名称 | 简介 |
| torch | PyTorch是一个灵活而强大的深度学习库,具有易于部署、分布式训练和强大的生态系统等功能。 |
| torchaudio | PyTorch的音频包 |
| torchdata | PyTorch的可组合数据加载模块 |
| torchtext | PyTorch的文本实用程序和数据集 |
| torchvision | torch深度学习的图像和视频数据集及模型 |
其他包
下载地址:download.pytorch.org/whl/cu124
| =名称 | =简介 |
|---|---|
| torch | PyTorch是一个灵活而强大的深度学习库,具有易于部署、分布式训练和强大的生态系统等功能。 |
| torch-cuda80 | PyNVVL:使用CuPy的NVIDIA视频加载器(NVVL)的Python包装 |
| torch-model-archiver | Torch Model Archiver用于创建可由TorchServe推理使用的经过训练的神经网络模型的档案 |
| torch-tb-profiler | PyTorch Profiler TensorBoard插件 |
| torch-tensorrt | Torch-TensorRT 是 NVIDIA 和 Pytorch 合作推出的一个编译器,支持将 PyTorch/TorchScript 模型直接编译后使用 TensorRT 部署。 |
| torchao | torchao通过利用低比特数据类型、量化和稀疏性来使模型更快、更小。 |
| torchaudio | PyTorch的音频包 |
| torchcodec | TorchCodec 是一个用于将视频解码为 PyTorch 张量的 Python 库,支持在 CPU 和 CUDA GPU 上运行。 |
| torchcsprng | PyTorch的加密安全伪随机数生成器 |
| torchdata | PyTorch的可组合数据加载模块 |
| torchmetrics | TorchMetrics是一个开源的PyTorch原生的函数和度量模块的集合,用于简单的性能评估。 |
| torchrec | 用于推荐系统的Pytorch域库 |
| torchrec-cpu | 用于推荐系统的Pytorch域库(CPU版) |
| torchserve | TorchServe是一种为神经网络模型提供推理服务的工具 |
| torchtext | PyTorch的文本实用程序和数据集 |
| torchtune | torchtune 是一个 PyTorch 库,用于轻松创作、微调和试验 LLM。 |
| torchvision | torch深度学习的图像和视频数据集及模型 |
最后更新于1年前
本文由人工编写,AI优化,转载请注明原文地址: PyTorch官方包简介
推荐阅读
使用Cesium.js加载vtu格式(UnstructuredGrid)的文件
162025-12-06
程序员常用技术大全:从编程语言到开发工具完整指南
4922025-10-19
IntelliJ IDEA常见问题解决方案大全:服务面板、Maven报错、启动故障处理
3422026-04-14
OpenVPN安装配置完整指南:从零搭建安全VPN服务器与客户端
22482025-10-30
PyTorch深度学习教程:从入门到实战与常见问题解答
4472025-10-31
Windows系统PyTorch安装教程:CUDA 12.1环境配置与TorchText版本兼容性指南
24442025-10-08