nnabla

本项目已于 [2025年4月3日] 停止维护。 不再提供任何开发或安全支持。

神经网络 库

神经网络 库 是一个 深度学习 框架，旨在用于研究、开发和 生产环境。我们的目标是让它运行在任何地方：桌面电脑、高性能计算 集群、嵌入式设备以及生产服务器。

• 神经网络 库 - CUDA 扩展：神经网络 库的扩展库，允许用户在使用 CUDA 的 GPU 上加速计算。
• 神经网络 库 - 示例：从基础到前沿的 神经网络 库实用示例。
• 神经网络 库 - C 运行时：由 神经网络 库创建的用于推理 神经网络 的运行时库。
• 神经网络 库 - NAS：面向 神经网络 库的硬件感知神经架构搜索（NAS）。
• 神经网络 库 - RL：基于 神经网络 库构建的深度强化学习（RL）库。
• 神经网络 控制台：用于 神经网络 开发的 Windows GUI 应用程序。

安装

安装 神经网络 库非常简单：

pip install nnabla

这会安装 神经网络 库的 CPU 版本。通过以下命令安装 CUDA 扩展，可以添加 GPU 加速支持。

pip install nnabla-ext-cuda116

上述命令适用于 CUDA 工具包 11.6 版本。

其他受支持的 CUDA 包列在此处。

目前不支持 CUDA 10.x、9.x 和 8.x 版本。

更多详情，请参阅文档中的安装部分。

从源码构建

请参阅构建手册。

在 Docker 上运行

关于在 Docker 上运行的详细信息，请参阅安装部分。