primitiv: A Neural Network Toolkit. (Python frontend)
Project description
Features
Dynamic and incremental graph construction
On-demand memory allocation
Automatic minibatch broadcasting
Mostly device-independent
Simple usage
Install
Prerequisites:
Python 3 (3.5 or later)
NumPy (1.11.0 or later)
Cython (0.27 or later)
CMake (3.1.0 or later)
scikit-build (0.6.1 or later)
(optional) CUDA (7.5 or later)
(optional) OpenCL (1.2 or later) and OpenCL C++ binding v2
Install required packages:
pip3 install numpy cython cmake scikit-build
Build and install primitiv without CUDA and OpenCL:
pip3 install primitiv
Build and install primitiv with CUDA and/or OpenCL support:
# 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
Notes
We are providing only a source pacakge for now, and pip command downloads the source package and builds it before installing. This is mainly because of keeping compatibility with the manylinux1 standard described in PEP 513 while maintaining supports of non-standard backends such as CUDA/OpenCL.
Resources
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.