fix ci

Summary:
X-link: fairinternal/detectron2#605

Pull Request resolved: #5483

Reviewed By: vbourgin

Differential Revision: D80125449

fbshipit-source-id: a7e8b8cc42e063b35b0d2ba64ad26bc99b674d61

Author: Yanghan Wang

.github/actions/install_linux_gpu_dep/action.yml

@@ -14,5 +14,6 @@ runs:
       # Installing the keyring seems to be unnecessary
 
       sudo apt-get update
+      apt-cache search cuda-toolkit
       sudo apt-get --yes install cuda-toolkit-${{inputs.cuda-version}}
       sudo apt-get --yes install nvidia-gds-${{inputs.cuda-version}}

.github/workflows/workflow.yml

@@ -72,7 +72,7 @@ jobs:
           # install from github to get latest; install iopath first since fvcore depends on it
           python -m pip install -U 'git+https://github.com/facebookresearch/iopath'
           python -m pip install -U 'git+https://github.com/facebookresearch/fvcore'
-          wget https://raw.githubusercontent.com/pytorch/pytorch/master/torch/utils/collect_env.py
+          wget https://raw.githubusercontent.com/pytorch/pytorch/v${{matrix.torch}}/torch/utils/collect_env.py
           python collect_env.py
 
       - name: Build and install
@@ -90,16 +90,13 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        torch: ["1.13.1", "2.2.2"]
+        torch: ["2.4.1"]
         include:
-          - torch: "1.13.1"
-            torchvision: "0.14.1"
-            cuda: "11-7"
-          - torch: "2.2.2"
-            torchvision: "0.17.2"
-            cuda: "12-5"
+          - torch: "2.4.1"
+            torchvision: "0.19.1"
+            cuda: "12-8"
     env:
-      PYTORCH_INDEX: "https://download.pytorch.org/whl/cu118"
+      PYTORCH_INDEX: "https://download.pytorch.org/whl/cu128"
       DETECTRON2_DATASETS: ~/.torch/datasets
     steps:
       - name: Checkout