@@ -39,37 +39,35 @@ jobs:
3939 run : |
4040 pip install --upgrade pip
4141
42- # - name: Install CUDA ${{ matrix.cuda-version }}
43- # if: ${{ matrix.cuda-version != 'cpu' }}
44- # run: |
45- # bash .github/workflows/cuda/${{ matrix.cuda-version }}-${{ runner.os }}.sh
42+ - name : Install CUDA ${{ matrix.cuda-version }}
43+ if : ${{ matrix.cuda-version != 'cpu' }}
44+ run : |
45+ bash .github/workflows/cuda/${{ matrix.cuda-version }}-${{ runner.os }}.sh
4646
4747 - name : Install PyTorch ${{ matrix.torch-version }}+${{ matrix.cuda-version }}
4848 run : |
4949 pip install numpy typing-extensions
50- pip install --no-index --no-cache-dir --verbose torch==${{ matrix.torch-version}} -f https://download.pytorch.org/whl/${{ matrix.cuda-version }}/torch_stable.html
51- python -c "import torch; print('PyTorch:', torch.__version__)"
52- python -c "import torch; print('CUDA: ', torch.version.cuda)"
50+ pip install --no-index --no-cache-dir torch==${{ matrix.torch-version}} -f https://download.pytorch.org/whl/${{ matrix.cuda-version }}/torch_stable.html
5351
54- # - name: Install main package for CPU
55- # if: ${{ matrix.cuda-version == 'cpu' }}
56- # run: |
57- # FORCE_ONLY_CPU=1 pip install -e .
58- # shell:
59- # bash
52+ - name : Install main package for CPU
53+ if : ${{ matrix.cuda-version == 'cpu' }}
54+ run : |
55+ FORCE_ONLY_CPU=1 pip install -e .
56+ shell :
57+ bash
6058
61- # - name: Install main package for GPU
62- # if: ${{ matrix.cuda-version != 'cpu' }}
63- # run: |
64- # source .github/workflows/cuda/${{ matrix.cuda-version }}-${{ runner.os }}-env.sh
65- # FORCE_CUDA=1 pip install -e .
66- # shell:
67- # bash
59+ - name : Install main package for GPU
60+ if : ${{ matrix.cuda-version != 'cpu' }}
61+ run : |
62+ source .github/workflows/cuda/${{ matrix.cuda-version }}-${{ runner.os }}-env.sh
63+ FORCE_CUDA=1 pip install -e .
64+ shell :
65+ bash
6866
69- # - name: Build wheel
70- # run: |
71- # pip install wheel
72- # python setup.py bdist_wheel --dist-dir=dist
67+ - name : Build wheel
68+ run : |
69+ pip install wheel
70+ python setup.py bdist_wheel --dist-dir=dist
7371
7472 # - name: Upload wheel
7573 # uses: shallwefootball/s3-upload-action@master
0 commit comments