文件名称:nnabla:神经网络库
文件大小:4.03MB
文件格式:ZIP
更新时间:2024-02-28 22:47:03
Python
神经网络库 神经网络库是一个深度学习框架,旨在用于研究,开发和生产。 我们的目标是使其在任何地方运行:台式机,HPC群集,嵌入式设备和生产服务器。 :的扩展库,允许用户在具有CUDA功能的GPU上加快计算速度。 :从基本到最新的神经网络库的工作示例。 :神经网络库创建的用于推断神经网络的运行时库。 :用于神经网络库的硬件感知的神经体系结构搜索(NAS)。 :用于神经网络开发的Windows GUI应用程序。 安装 安装神经网络库很容易: pip install nnabla 这将安装神经网络库的CPU版本。 可以通过使用以下命令安装CUDA扩展来添加GPU加速。 pip install nnabla-ext-cuda101 上面的命令适用于版本10.1 CUDA工具包。 对于其他版本: 为CUDA版本10.0 pip install nnabla-ext-cuda100