testing with another container

xiwang0818 · Apr 20, 2022 · 2394341 · 2394341
1 parent da4dfcc
commit 2394341
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/.github/workflows/cml.yaml b/.github/workflows/cml.yaml
@@ -3,18 +3,19 @@ on: [push]
 jobs:
   run:
     runs-on: [ubuntu-latest]
-    container: docker://dvcorg/cml-py3:latest
+    #container: docker://dvcorg/cml-py3:latest
+    container: docker://iterativeai/cml:0-dvc2-base1-gpu
     steps:
       - uses: actions/checkout@v2
       - name: cml_run
         env:
           repo_token: ${{ secrets.GITHUB_TOKEN }}
         run: |
           # Your ML workflow goes here
-          pip install --upgrade pip
-          pip install torch torchvision torchaudio
-          pip install torch-scatter torch-sparse -f https://data.pyg.org/whl/torch-1.10.1+cpu.html
-          pip install -r requirements.txt
+          # pip install --upgrade pip
+          # pip install torch torchvision torchaudio
+          # pip install torch-scatter torch-sparse -f https://data.pyg.org/whl/torch-1.10.1+cpu.html
+          pip install -r requirements-pip.txt
           echo "CREATING DATA"
           python create_data.py
           echo "TRAINING THE MODEL"