diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index 5a19c7e..9c1014a 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build.yml @@ -82,25 +82,25 @@ jobs: pip install wheel setuptools packaging export TORCH_CUDA_VERSION=$(python -c " \ from os import environ as env; \ - support_cuda_versions = { - '2.0': [117, 118], - '2.1': [118, 121], - '2.2': [118, 121], - '2.3': [118, 121], - '2.4': [118, 121, 124], - '2.5': [118, 121, 124], + support_cuda_versions = { \ + '2.0': [117, 118], \ + '2.1': [118, 121], \ + '2.2': [118, 121], \ + '2.3': [118, 121], \ + '2.4': [118, 121, 124], \ + '2.5': [118, 121, 124], \ }; \ target_cuda_versions = support_cuda_versions[env['MATRIX_TORCH_VERSION']]; \ - cuda_version = env['MATRIX_CUDA_VERSION']; \ - closest_version = min(target_cuda_versions, key=lambda x: abs(x - int(cuda_version))); \ + cuda_version = int(env['MATRIX_CUDA_VERSION']); \ + closest_version = min(target_cuda_versions, key=lambda x: abs(x - cuda_version)); \ print(closest_version) \ - ) + ") pip install --no-cache-dir torch==${{ matrix.torch }} --index-url https://download.pytorch.org/whl/cu${TORCH_CUDA_VERSION} nvcc --version python --version python -c "import torch; print('PyTorch:', torch.__version__)" python -c "import torch; print('CUDA:', torch.version.cuda)" - python -c "from torch.utils import cpp_extension; print (cpp_extension.CUDA_HOME)" + python -c "from torch.utils import cpp_extension; print(cpp_extension.CUDA_HOME)" - name: Build run: | diff --git a/torch_cpp/__init__.py b/torch_cpp/__init__.py index 09f03d7..234be32 100644 --- a/torch_cpp/__init__.py +++ b/torch_cpp/__init__.py @@ -1,3 +1,3 @@ from .src import * # noqa: F401, F403 -__version__ = "1.1.6" +__version__ = "1.1.7"