Commit 9c9c1467 authored by rusty1s's avatar rusty1s
Browse files

[skip ci] bash shell

parent ca99ecfd
...@@ -46,12 +46,11 @@ jobs: ...@@ -46,12 +46,11 @@ jobs:
- name: Install PyTorch ${{ matrix.torch-version }}+${{ matrix.cuda-version }} - name: Install PyTorch ${{ matrix.torch-version }}+${{ matrix.cuda-version }}
run: | run: |
pip install --no-cache-dir torch==${{ matrix.torch-version}} -f https://download.pytorch.org/whl/torch_stable/+${{ matrix.cuda-version }}.html pip install --no-cache-dir torch==${{ matrix.torch-version}} -f https://download.pytorch.org/whl/${{ matrix.cuda-version }}/torch_stable.html
- name: Install main package for CPU - name: Install main package for CPU
if: ${{ matrix.cuda-version == 'cpu' }} if: ${{ matrix.cuda-version == 'cpu' }}
run: | run: |
python -c "import torch; print(torch.__version__ + '+cpu')"
FORCE_ONLY_CPU=1 pip install -e . FORCE_ONLY_CPU=1 pip install -e .
shell: shell:
bash bash
...@@ -59,7 +58,6 @@ jobs: ...@@ -59,7 +58,6 @@ jobs:
- name: Install main package with CUDA - name: Install main package with CUDA
if: ${{ matrix.cuda-version != 'cpu' }} if: ${{ matrix.cuda-version != 'cpu' }}
run: | run: |
python -c "import torch; print(torch.__version__ + '+' + torch.version.cuda)"
source .github/workflows/cuda/${{ matrix.cuda-version }}-${{ runner.os }}-env.sh source .github/workflows/cuda/${{ matrix.cuda-version }}-${{ runner.os }}-env.sh
FORCE_CUDA=1 pip install -e . FORCE_CUDA=1 pip install -e .
shell: shell:
......
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