зеркало из https://github.com/microsoft/DeepSpeed.git
Update setup.py (#298)
This commit is contained in:
Родитель
1f972427fc
Коммит
871f7e6305
2
setup.py
2
setup.py
|
@ -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])
|
||||
|
|
Загрузка…
Ссылка в новой задаче