diff --git a/.github/workflows/pythonapp-gpu.yml b/.github/workflows/pythonapp-gpu.yml index 1b1a484105..d19bbe437f 100644 --- a/.github/workflows/pythonapp-gpu.yml +++ b/.github/workflows/pythonapp-gpu.yml @@ -23,7 +23,7 @@ jobs: - "PT17+CUDA102" - "PT18+CUDA102" - "PT18+CUDA112" - - "PT110+CUDA116" + - "PT111+CUDA116" - "PT110+CUDA102" - "PT111+CUDA102" include: @@ -44,7 +44,7 @@ jobs: # 21.10: 1.10.0a0+0aef44c pytorch: "-h" base: "nvcr.io/nvidia/pytorch:21.10-py3" - - environment: PT110+CUDA116 + - environment: PT111+CUDA116 # we explicitly set pytorch to -h to avoid pip install error # 22.03: 1.12.0a0+2c916ef pytorch: "-h"