PyTorch简介

创建日期:2024-06-21
更新日期:2025-03-11

官方包

名称简介
torchPyTorch是一个灵活而强大的深度学习库,具有易于部署、分布式训练和强大的生态系统等功能。
torchaudioPyTorch的音频包
torchdataPyTorch的可组合数据加载模块
torchtextPyTorch的文本实用程序和数据集
torchvisiontorch深度学习的图像和视频数据集及模型

其他包

下载地址:download.pytorch.org/whl/cu124

=名称=简介
torchPyTorch是一个灵活而强大的深度学习库,具有易于部署、分布式训练和强大的生态系统等功能。
torch-cuda80PyNVVL:使用CuPy的NVIDIA视频加载器(NVVL)的Python包装
torch-model-archiverTorch Model Archiver用于创建可由TorchServe推理使用的经过训练的神经网络模型的档案
torch-tb-profilerPyTorch Profiler TensorBoard插件
torch-tensorrtTorch-TensorRT 是 NVIDIA 和 Pytorch 合作推出的一个编译器,支持将 PyTorch/TorchScript 模型直接编译后使用 TensorRT 部署。
torchaotorchao通过利用低比特数据类型、量化和稀疏性来使模型更快、更小。
torchaudioPyTorch的音频包
torchcodecTorchCodec 是一个用于将视频解码为 PyTorch 张量的 Python 库,支持在 CPU 和 CUDA GPU 上运行。
torchcsprngPyTorch的加密安全伪随机数生成器
torchdataPyTorch的可组合数据加载模块
torchmetricsTorchMetrics是一个开源的PyTorch原生的函数和度量模块的集合,用于简单的性能评估。
torchrec用于推荐系统的Pytorch域库
torchrec-cpu用于推荐系统的Pytorch域库(CPU版)
torchserveTorchServe是一种为神经网络模型提供推理服务的工具
torchtextPyTorch的文本实用程序和数据集
torchtunetorchtune 是一个 PyTorch 库,用于轻松创作、微调和试验 LLM。
torchvisiontorch深度学习的图像和视频数据集及模型