未验证 提交 871f7e63 编写于 作者: J Jeff Rasley 提交者: GitHub

Update setup.py (#298)

上级 1f972427
......@@ -14,7 +14,7 @@ from setuptools import setup, find_packages
from torch.utils.cpp_extension import CUDAExtension, BuildExtension
cmdclass = {}
cmdclass['build_ext'] = BuildExtension
cmdclass['build_ext'] = BuildExtension.with_options(use_ninja=False)
TORCH_MAJOR = int(torch.__version__.split('.')[0])
TORCH_MINOR = int(torch.__version__.split('.')[1])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册