nexus-torch is a tensor computation backend for Nexus based on PyTorch.
It uses SWIG to generate the JNI bindings directly from the Torch C/C++ source without the Python part of PyTorch.
- Linux + CUDA 9.1 / 10.0 + PyTorch 1.0.0
Building the binding manually
- PyTorch 1.0.0
pip install torch==1.0.0
- SWIG 3.0+
- Patch SWIG.
swig-patch/fix-long.patchfile to patch SWIG: This resolves SWIG issue #646, which incorrectly maps
Intinstead of the correct
Longunder 64-bit machines.
- Generate JNI shared library: Run
build.shto build the shared library (