Explicit PyTorch installation for MacOS

ccb-hms · Sep 15, 2024 · 5cf406e · 5cf406e
1 parent 5b1e077
commit 5cf406e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/.github/workflows/pytest.yml b/.github/workflows/pytest.yml
@@ -27,11 +27,14 @@ jobs:
         with:
           python-version: ${{ matrix.python-version }}
 
-      - name: install dependencies
+      - name: install Pipenv, Pipfile
         run: |
           python -m pip install pipenv
           pipenv install --dev --ignore-pipfile --python=${{ matrix.python-version }}
 
+      - name: install PyTorch
+        run: pipenv run python -m pip install torch torchvision torchaudio
+
       - name: install detectron2
         run: bash ./bash_scripts/install_detectron