mirrors/mim
1
0
Fork 0
mirror of https://github.com/open-mmlab/mim.git synced 2025-06-03 14:59:11 +08:00
Code Issues Projects Releases Wiki Activity

add pytorch1.5 in ci

Browse Source
This commit is contained in:
zhouzaida 2021-05-20 17:45:04 +08:00
parent cea7f5f41e
commit 26dc15068f
1 changed files with 18 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
.github/workflows/build.yml vendored
View File

@ -29,13 +29,27 @@ jobs:
CUDA_SHORT: 10.1
UBUNTU_VERSION: ubuntu1804
FORCE_CUDA: 1
strategy:
matrix:
python-version: [3.7]
torch: [1.5.0+cu101, 1.8.0+cu101]
include:
- torch: 1.5.0+cu101
torchvision: 0.6.0+cu101
- torch: 1.8.0+cu101
torchvision: 0.9.0+cu101
- torch: 1.8.0+cu101
torchvision: 0.9.0+cu101
python-version: 3.6
- torch: 1.8.0+cu101
torchvision: 0.9.0+cu101
python-version: 3.8
steps:
- uses: actions/checkout@v2
- name: Set up Python 3.7
- name: Set up Python ${{ matrix.python-version }}
uses: actions/setup-python@v2
with:
python-version: 3.7
python-version: ${{ matrix.python-version }}
- name: Install CUDA
run: |
export INSTALLER=cuda-repo-${UBUNTU_VERSION}_${CUDA}_amd64.deb
@ -50,7 +64,7 @@ jobs:
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${CUDA_HOME}/include:${LD_LIBRARY_PATH}
export PATH=${CUDA_HOME}/bin:${PATH}
- name: Install PyTorch
run: pip install torch==1.8.0+cu101 torchvision==0.9.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html
run: pip install torch==${{matrix.torch}} torchvision==${{matrix.torchvision}} -f https://download.pytorch.org/whl/torch_stable.html
- name: Build and install
run: rm -rf .eggs && pip install -e .
- name: Validate the installation