Commit 89cfdcfd authored by rusty1s's avatar rusty1s
Browse files

[skip ci] build pytorch 1.9.0

parent f9ee2308
...@@ -44,7 +44,7 @@ jobs: ...@@ -44,7 +44,7 @@ jobs:
- name: Install PyTorch ${{ matrix.torch-version }}+${{ matrix.cuda-version }} - name: Install PyTorch ${{ matrix.torch-version }}+${{ matrix.cuda-version }}
run: | run: |
pip install numpy typing-extensions pip install numpy typing-extensions dataclasses
pip install --no-index --no-cache-dir torch==${{ matrix.torch-version}} -f https://download.pytorch.org/whl/${{ matrix.cuda-version }}/torch_stable.html pip install --no-index --no-cache-dir torch==${{ matrix.torch-version}} -f https://download.pytorch.org/whl/${{ matrix.cuda-version }}/torch_stable.html
python -c "import torch; print('PyTorch:', torch.__version__)" python -c "import torch; print('PyTorch:', torch.__version__)"
python -c "import torch; print('CUDA:', torch.version.cuda)" python -c "import torch; print('CUDA:', torch.version.cuda)"
......
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