加速PyTorch研究和开发的补充组件。
项目描述
pytorch-pfn-extras
加速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 查看散列)
关闭
pytorch_pfn_extras-0.7.7.tar.gz的散列
算法 | 散列摘要 | |
---|---|---|
SHA256 | 11164018ca8bb7fccd6cfcc392ce28408f51c54d8fd38ad5d8a7322500d58622 |
|
MD5 | 3558012d7b564d0409550ffe93cf8328 |
|
BLAKE2b-256 | d832123d73d7ba0550677676fd4da1f1e63a93413d4e908b4b6c97f09474e373 |
关闭
pytorch_pfn_extras-0.7.7-py3-none-any.whl的散列
算法 | 散列摘要 | |
---|---|---|
SHA256 | 6ea32a97b22c879461200f4792bbcf15174138aed41d386a22030c16d9d2a404 |
|
MD5 | 203b8f3afd12cd8181818a2dc92f4a87 |
|
BLAKE2b-256 | 54526c361bdbb32419cacf99e9b8135249a5acd821d60c116333a9d3402e1c02 |