diff --git a/.github/workflows/ci-pip-install-and-test-gpu.yml b/.github/workflows/ci-pip-install-and-test-gpu.yml index b05b2ecf..dab7b060 100644 --- a/.github/workflows/ci-pip-install-and-test-gpu.yml +++ b/.github/workflows/ci-pip-install-and-test-gpu.yml @@ -13,6 +13,11 @@ jobs: - name: Checkout uses: actions/checkout@v2 + - name: Set up Python 3.9 + uses: actions/setup-python@v2 + with: + python-version: 3.9 + - name: Install torch (GPU CUDA 12.1) run: | python -m pip install torch --index-url https://download.pytorch.org/whl/cu121