Commit b0a685f3 authored by rusty1s's avatar rusty1s
Browse files

ci-deploy PyTorch 1.8.0 wheels

parent 69d8bcf6
......@@ -14,31 +14,31 @@ env:
- PYTHON_VERSION=3.6 TORCH_VERSION=1.8.0 IDX=cu101
- PYTHON_VERSION=3.6 TORCH_VERSION=1.8.0 IDX=cu102
- PYTHON_VERSION=3.6 TORCH_VERSION=1.8.0 IDX=cu111
- PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cpu
- PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cu92
- PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cu101
- PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cu102
- PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cu110
# - PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cpu
# - PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cu92
# - PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cu101
# - PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cu102
# - PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cu110
# Python 3.7
- PYTHON_VERSION=3.7 TORCH_VERSION=1.8.0 IDX=cpu
- PYTHON_VERSION=3.7 TORCH_VERSION=1.8.0 IDX=cu101
- PYTHON_VERSION=3.7 TORCH_VERSION=1.8.0 IDX=cu102
- PYTHON_VERSION=3.7 TORCH_VERSION=1.8.0 IDX=cu111
- PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cpu
- PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cu92
- PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cu101
- PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cu102
- PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cu110
# - PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cpu
# - PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cu92
# - PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cu101
# - PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cu102
# - PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cu110
# Python 3.8
- PYTHON_VERSION=3.8 TORCH_VERSION=1.8.0 IDX=cpu
- PYTHON_VERSION=3.8 TORCH_VERSION=1.8.0 IDX=cu101
- PYTHON_VERSION=3.8 TORCH_VERSION=1.8.0 IDX=cu102
- PYTHON_VERSION=3.8 TORCH_VERSION=1.8.0 IDX=cu111
- PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cpu
- PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cu92
- PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cu101
- PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cu102
- PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cu110
# - PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cpu
# - PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cu92
# - PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cu101
# - PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cu102
# - PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cu110
jobs:
# For daily builds, we only test on Python 3.6 with 11.0/11.1.
......
......@@ -10,9 +10,6 @@ from torch.__config__ import parallel_info
from torch.utils.cpp_extension import BuildExtension
from torch.utils.cpp_extension import CppExtension, CUDAExtension, CUDA_HOME
major, minor = int(str(torch.__version__)[0]), int(str(torch.__version__)[2])
use_ninja = major >= 2 or (major == 1 and minor >= 8)
WITH_CUDA = torch.cuda.is_available() and CUDA_HOME is not None
suffices = ['cpu', 'cuda'] if WITH_CUDA else ['cpu']
if os.getenv('FORCE_CUDA', '0') == '1':
......@@ -105,8 +102,7 @@ setup(
ext_modules=get_extensions() if not BUILD_DOCS else [],
cmdclass={
'build_ext':
BuildExtension.with_options(no_python_abi_suffix=True,
use_ninja=use_ninja)
BuildExtension.with_options(no_python_abi_suffix=True, use_ninja=False)
},
packages=find_packages(),
)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment