跳转到主要内容

加速PyTorch研究和开发的补充组件。

项目描述

pytorch-pfn-extras

PyPI Docs License

加速PyTorch研究和开发的补充组件。

安装

pip install pytorch-pfn-extras

# Use `[onnx]` to use onnx submodule like:
#  pip install "pytorch-pfn-extras[onnx]"

### Optinal dependencies
# For PlotReport / VariableStatisticsPlot extensions
pip install matplotlib

# For IgniteExtensionsManager
pip install pytorch-ignite torchvision

# For CuPy interoperability (see: https://docs.cupy.dev/en/stable/install.html)
pip install cupy  # or cupy-cudaXXX

需求

  • Python 3.8+
  • PyTorch 1.10+

可选依赖

  • CuPy 8.0+ 用于 PyTorch/CuPy 互操作性

文档

请参阅Read The Docs以获取完整文档。

以下是到库最重要功能的快速链接。

示例

贡献指南

您可以通过发送拉取请求来为此项目做出贡献。经审核批准后,审阅者将合并拉取请求。

在做出贡献之前,请确认以下内容

  • 代码质量在整个脚本、模块或包中保持一致。
  • 代码被单元测试覆盖。
  • API易于维护。

许可证

MIT许可证

项目详情


下载文件

下载适用于您的平台的文件。如果您不确定选择哪个,请了解更多关于安装包的信息。

源代码分发

pytorch_pfn_extras-0.7.7.tar.gz (162.5 kB 查看散列)

上传时间 源代码

构建分发

pytorch_pfn_extras-0.7.7-py3-none-any.whl (225.6 kB 查看散列)

上传时间 Python 3

支持者