Merge pull request #75 from FR-DC/frml-159

FRML-159 Add torch with gpu to Poetry
FR-DC · Jun 19, 2024 · d5bc1f1 · d5bc1f1
2 parents 730fdc4 + 110c253
commit d5bc1f1
Show file tree

Hide file tree

Showing 4 changed files with 1,046 additions and 191 deletions.
diff --git a/.github/workflows/basic-tests.yml b/.github/workflows/basic-tests.yml
@@ -6,6 +6,7 @@ on:
   pull_request:
     paths:
       - "src/frdc/**"
+      - "poetry.lock"
 
 jobs:
   build:
@@ -40,7 +41,6 @@ jobs:
           python -m pip install flake8 pytest poetry          
           poetry export --with dev --without-hashes -o requirements.txt
           pip install -r requirements.txt
-          pip install torch torchaudio torchvision lightning
 
       - name: Lint with flake8
         run: |

diff --git a/.github/workflows/model-tests.yml b/.github/workflows/model-tests.yml
@@ -53,7 +53,6 @@ jobs:
           python3 -m pip install flake8 pytest poetry
           poetry export --with dev --without-hashes -o requirements.txt
           pip3 install -r requirements.txt
-          pip3 install torch torchvision torchaudio
 
       - name: Check CUDA is available
         run: nvidia-smi