pytorch/setup.py
2016-05-02 23:19:57 +02:00

11 lines
320 B
Python

from setuptools import setup, Extension
C = Extension("torch.C",
libraries=['TH'],
sources=["torch/csrc/Module.c", "torch/csrc/Tensor.c", "torch/csrc/Storage.c"],
include_dirs=["torch/csrc"])
setup(name="torch", version="0.1",
ext_modules=[C],
packages=['torch'])