PRIMTIV:A Neural Network Toolkit.(Python Fronted)
primitiv的Python项目详细描述
功能
- 动态和增量图构造
- 按需分配内存
- 自动小批量广播
- 主要与设备无关
- 简单用法
安装
先决条件:
- python 3(3.5或更高版本)
- numpy(1.11.0或更高版本)
- cython(0.27或更高版本)
- SCIKIT构建(0.6.1或更高版本,仅用于构建)
- (可选)CUDA(7.5或更高版本)
- (可选)OpenCL(1.2或更高版本)和OpenCL C++绑定V2
安装依赖项:
pip3 install numpy cython scikit-build
不带CUDA和OpenCL安装Primitiv:
pip3 install primitiv
使用CUDA和/或OpenCL支持安装Primitiv:
# Enable only CUDA pip3 install primitiv --global-option --enable-cuda # Enable both CUDA and OpenCL pip3 install primitiv --global-option --enable-cuda --global-option --enable-opencl