switch CI to pip installs for torch, pyg and deps.

a-r-j · Feb 7, 2024 · 6d2f3d0 · 6d2f3d0
1 parent fc6bed8
commit 6d2f3d0
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
@@ -47,12 +47,14 @@ jobs:
       - name: Install mmseqs
         run: mamba install -c conda-forge -c bioconda mmseqs2
       - name: Install PyTorch
-        run: mamba install -c pytorch pytorch==${{matrix.torch}} cpuonly
-        #run: pip install torch==${{matrix.torch}}+cpu torchvision==${{matrix.torchvision}}+cpu -f https://download.pytorch.org/whl/torch_stable.html
+        #run: mamba install -c pytorch pytorch==${{matrix.torch}} cpuonly
+        run: pip install torch==${{matrix.torch}}+cpu torchvision==${{matrix.torchvision}}+cpu -f https://download.pytorch.org/whl/torch_stable.html
       - name: Install PyG
-        run: mamba install -c pyg pyg
+        #run: mamba install -c pyg pyg
+        run: pip install torch_geometric
       - name: Install torch-cluster
-        run: mamba install pytorch-cluster -c pyg
+        #run: mamba install pytorch-cluster -c pyg
+        run: pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-${{matrix.torch}}.html
       - name: Install BLAST
         run: sudo apt install ncbi-blast+
       - name: Install Graphein