官方包
名称 | 简介 |
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深度学习的图像和视频数据集及模型 |