Skip to main content

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


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

primitiv-0.4.0.dev159.tar.gz (185.4 kB view hashes)

Uploaded Source

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page