mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
11 lines
320 B
Python
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'])
|