fix install torch (#1247) #2
Workflow file for this run
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: build-wheels-cuda-ubuntu | |
| on: | |
| push: | |
| tags: | |
| - '*' | |
| workflow_dispatch: | |
| concurrency: | |
| group: build-wheels-cuda-ubuntu-${{ github.ref }} | |
| cancel-in-progress: true | |
| jobs: | |
| generate_build_matrix: | |
| if: github.repository_owner == 'csukuangfj' || github.repository_owner == 'k2-fsa' | |
| # see https://github.com/pytorch/pytorch/pull/50633 | |
| runs-on: ubuntu-latest | |
| outputs: | |
| matrix: ${{ steps.set-matrix.outputs.matrix }} | |
| steps: | |
| - uses: actions/checkout@v2 | |
| with: | |
| fetch-depth: 0 | |
| - name: Generating build matrix | |
| id: set-matrix | |
| run: | | |
| # outputting for debugging purposes | |
| python ./scripts/github_actions/generate_build_matrix.py --enable-cuda | |
| MATRIX=$(python ./scripts/github_actions/generate_build_matrix.py --enable-cuda) | |
| echo "::set-output name=matrix::${MATRIX}" | |
| build-manylinux-wheels: | |
| needs: generate_build_matrix | |
| name: ${{ matrix.torch }} ${{ matrix.python-version }} cuda${{ matrix.cuda }} | |
| runs-on: ubuntu-latest | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| ${{ fromJson(needs.generate_build_matrix.outputs.matrix) }} | |
| steps: | |
| - uses: actions/checkout@v2 | |
| with: | |
| fetch-depth: 0 | |
| - name: display | |
| shell: bash | |
| run: | | |
| # see https://github.com/pytorch/test-infra/blob/9e3d392690719fac85bad0c9b67f530e48375ca1/tools/scripts/generate_binary_build_matrix.py | |
| # https://github.com/pytorch/builder/tree/main/manywheel | |
| # https://github.com/pytorch/builder/pull/476 | |
| # https://github.com/k2-fsa/k2/issues/733 | |
| # https://github.com/pytorch/pytorch/pull/50633 (generate build matrix) | |
| - name: Run the build process with Docker | |
| uses: addnab/docker-run-action@v3 | |
| with: | |
| image: ${{ matrix.image }} | |
| options: -v ${{ github.workspace }}:/var/www -e PYTHON_VERSION=${{ matrix.python-version }} -e TORCH_VERSION=${{ matrix.torch }} -e CUDA_VERSION=${{ matrix.cuda }} | |
| run: | | |
| echo "pwd: $PWD" | |
| uname -a | |
| id | |
| cat /etc/*release | |
| gcc --version | |
| python3 --version | |
| which python3 | |
| pushd /usr/local | |
| rm cuda | |
| ln -s cuda-$CUDA_VERSION cuda | |
| popd | |
| which nvcc | |
| nvcc --version | |
| cp /var/www/scripts/github_actions/install_torch.sh . | |
| chmod +x install_torch.sh | |
| /var/www/scripts/github_actions/build-ubuntu-cuda.sh | |
| - name: Display wheels | |
| shell: bash | |
| run: | | |
| ls -lh ./wheelhouse/ | |
| - name: Upload Wheel | |
| uses: actions/upload-artifact@v2 | |
| with: | |
| name: torch-${{ matrix.torch }}-python-${{ matrix.python-version }}-ubuntu-latest-cuda | |
| path: wheelhouse/*.whl | |
| # https://huggingface.co/docs/hub/spaces-github-actions | |
| - name: Publish to huggingface | |
| if: github.repository_owner == 'csukuangfj' | |
| env: | |
| HF_TOKEN: ${{ secrets.HF_TOKEN }} | |
| uses: nick-fields/retry@v2 | |
| with: | |
| max_attempts: 20 | |
| timeout_seconds: 200 | |
| shell: bash | |
| command: | | |
| git config --global user.email "[email protected]" | |
| git config --global user.name "Fangjun Kuang" | |
| rm -rf huggingface | |
| export GIT_LFS_SKIP_SMUDGE=1 | |
| git clone https://huggingface.co/csukuangfj/k2 huggingface | |
| cd huggingface | |
| git pull | |
| mkdir -p ubuntu-cuda | |
| cp -v ../wheelhouse/*.whl ./ubuntu-cuda | |
| git status | |
| git lfs track "*.whl" | |
| git add . | |
| git commit -m "upload ubuntu-cuda wheel for torch ${{ matrix.torch }} python ${{ matrix.python-version }}" | |
| git push https://csukuangfj:[email protected]/csukuangfj/k2 main |