This commit is contained in:
Jeff Rasley 2020-07-22 08:33:05 -07:00 коммит произвёл GitHub
Родитель 1f972427fc
Коммит 871f7e6305
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -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])