MultiCUDA CI
@@ -100,6 +100,7 @@ jobs:
- "PT16+CUDA101"
- "PT15+CUDA102"
- "PT16+CUDA102"
+ - "PT16+CUDA110"
include:
- environment: PT15+CUDA101
pytorch: "torch==1.5.0+cu101 torchvision==0.6.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html"
@@ -113,6 +114,10 @@ jobs:
- environment: PT16+CUDA102
pytorch: "torch torchvision"
base: "nvcr.io/nvidia/cuda:10.2-devel-ubuntu18.04"
+ - environment: PT16+CUDA110
+ # we explicitly set pytorch to -h to avoid pip install error
+ pytorch: "-h"
+ base: "nvcr.io/nvidia/pytorch:20.07-py3"
container:
image: ${{ matrix.base }}
options: --gpus all
@@ -121,6 +126,7 @@ jobs:
- uses: actions/checkout@v2
- name: apt install
run: |
+ if [ ${{ matrix.environment }} != "PT16+CUDA110" ]; then \
PYVER=3.6 PYSFX=3 DISTUTILS=python3-distutils && \
apt-get update && apt-get install -y --no-install-recommends \
curl \
@@ -150,7 +156,7 @@ jobs:
ln -s /usr/bin/python$PYVER /usr/bin/python`echo $PYVER | cut -c1-1` &&
curl -O https://bootstrap.pypa.io/get-pip.py && \
python get-pip.py && \
- rm get-pip.py
+ rm get-pip.py ; fi
- name: Install dependencies
run: |
which python
