Merge branch 'main' into windows-support #22
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: Win Test | |
| on: push | |
| jobs: | |
| build-wheels: | |
| name: Build Wheels | |
| runs-on: windows-latest | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| python-version: [3.6, 3.7, 3.8, 3.9] | |
| pytorch-version: [1.9, '1.10', 1.11, 1.12] | |
| cuda-version: [10.2, 11.3, 11.6] | |
| include: | |
| - cuda-version: 10.2 | |
| cuda-minor-version: 89 #https://developer.download.nvidia.com/compute/cuda/10.2/Prod/network_installers/cuda_10.2.89_win10_network.exe | |
| cuda-arch-list: "3.7 5.0 6.0 6.1 6.2 7.0 7.2 7.5+PTX" | |
| gcc-version: 8 | |
| - cuda-version: 11.3 | |
| cuda-minor-version: 0 #https://developer.download.nvidia.com/compute/cuda/11.3.0/network_installers/cuda_11.3.0_win10_network.exe | |
| cuda-arch-list: "3.7 5.0 6.0 6.1 6.2 7.0 7.2 7.5 8.0 8.6+PTX" | |
| gcc-version: 10 | |
| - cuda-version: 11.6 | |
| cuda-minor-version: 0 #https://developer.download.nvidia.com/compute/cuda/11.6.0/network_installers/cuda_11.6.0_windows_network.exe | |
| cuda-arch-list: "3.7 5.0 6.0 6.1 6.2 7.0 7.2 7.5 8.0 8.6+PTX" | |
| gcc-version: 10 | |
| exclude: | |
| - python-version: 3.6 | |
| pytorch-version: 1.11 | |
| - python-version: 3.6 | |
| pytorch-version: 1.12 | |
| - pytorch-version: 1.9 | |
| cuda-version: 11.3 | |
| - pytorch-version: 1.9 | |
| cuda-version: 11.6 | |
| - pytorch-version: '1.10' | |
| cuda-version: 11.6 | |
| - pytorch-version: 1.11 | |
| cuda-version: 11.6 | |
| steps: | |
| - name: Install Python | |
| uses: actions/setup-python@v4 | |
| with: | |
| python-version: ${{ matrix.python-version }} | |
| - name: Install MSVC | |
| uses: ilammy/msvc-dev-cmd@v1 | |
| - name: Install Cuda | |
| uses: Jimver/[email protected] | |
| id: cuda-toolkit | |
| with: | |
| cuda: '${{ matrix.cuda-version }}.${{ matrix.cuda-minor-version }}' | |
| - name: Install Pytorch | |
| run: | | |
| FULL_PYTORCH_VERSION=$(python -m pip index versions torch -f https://download.pytorch.org/whl/torch_stable.html | grep -o ${PYTORCH_VERSION}.[0-9]+cu${CUDA_VERSION//.} | head -n 1); \ | |
| python -m pip install torch==${FULL_PYTORCH_VERSION} -f https://download.pytorch.org/whl/torch_stable.html; | |
| python -m pip install numpy wheel ninja | |
| - name: Checkout Repository | |
| uses: actions/checkout@v2 | |
| - name: Build Wheels | |
| working-directory: test_extension | |
| run: python setup.py install |