use conda instead

Author: SkafteNicki

.github/workflows/tests.yml

@@ -14,23 +14,33 @@ jobs:
       matrix:
         os: [ubuntu-20.04, macOS-10.15, windows-2019]
         python-version: [3.6, 3.7, 3.8, 3.9]
-        pytorch-version: ["1.6", "1.7", "1.8", "1.9", "1.10", "1.11"]
+        pytorch-version: ["1.7", "1.8", "1.9", "1.10", "1.11"]
 
     steps:
     - name: Checkout
       uses: actions/checkout@v2
 
-    - name: Set up Python ${{ matrix.python-version }}
-      uses: actions/setup-python@v2
+    - name: Setup Miniconda using Python ${{ matrix.python-version }}
+      uses: conda-incubator/setup-miniconda@v2
       with:
+        miniforge-variant: Mambaforge
+        miniforge-version: latest
+        use-mamba: true
         python-version: ${{ matrix.python-version }}
+        channels: pytorch,pytorch-test,pytorch-nightly,conda-forge
+        channel-priority: true
+        use-only-tar-bz2: true
+      # prevent hanging Conda creations
+      timeout-minutes: 10
 
     - name: Install dependencies
       run: |
-        python -m pip install --upgrade pip
-        pip install torch==${{ matrix.pytorch-version }}
+        conda info
+        conda install cpuonly mkl pytorch=${{ matrix.pytorch-version }} packaging
         pip install -r requirements.txt
         pip install -r requirements_tests.txt
+        pip list
+        python -c "from torch import __version__ as ver; assert '.'.join(ver.split('.')[:2]) == '${{ matrix.pytorch-version }}', ver"
 
     - name: Install package
       run: |

requirements.txt

@@ -1,4 +1,4 @@
-torch>=1.6
+torch>=1.7
 numpy>=1.16.4
 torchplot
 networkx