diff --git a/.github/workflows/pythonapp-gpu.yml b/.github/workflows/pythonapp-gpu.yml index e290bcfd99..6e460a2ade 100644 --- a/.github/workflows/pythonapp-gpu.yml +++ b/.github/workflows/pythonapp-gpu.yml @@ -19,17 +19,13 @@ jobs: strategy: matrix: environment: - - "PT16+CUDA110" - "PT17+CUDA102" - "PT17+CUDA110" - "PT18+CUDA102" + - "PT18+CUDA110" - "PT19+CUDA114" - "PT110+CUDA102" include: - - environment: PT16+CUDA110 - # we explicitly set pytorch to -h to avoid pip install error - pytorch: "-h" - base: "nvcr.io/nvidia/pytorch:20.07-py3" - environment: PT17+CUDA102 pytorch: "torch==1.7.1 torchvision==0.8.2" base: "nvcr.io/nvidia/cuda:10.2-devel-ubuntu18.04" @@ -40,6 +36,10 @@ jobs: - environment: PT18+CUDA102 pytorch: "torch==1.8.1 torchvision==0.9.1" base: "nvcr.io/nvidia/cuda:10.2-devel-ubuntu18.04" + - environment: PT18+CUDA110 + # we explicitly set pytorch to -h to avoid pip install error + pytorch: "-h" + base: "nvcr.io/nvidia/pytorch:21.02-py3" - environment: PT19+CUDA114 # we explicitly set pytorch to -h to avoid pip install error # https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes