[skip ci] update cuda installation

.github/workflows/building.yml

@@ -40,26 +40,16 @@ jobs:
       - name: Install CUDA ${{ matrix.cuda-version }}
         if: ${{ matrix.cuda-version != 'cpu' }}
         run: |
-          source .github/workflows/${{ matrix.cuda-version }}-${{ runner.os }}.sh
+          source .github/workflows/cuda/${{ matrix.cuda-version }}-${{ runner.os }}.sh
 
       - name: Install PyTorch ${{ matrix.torch-version }}+${{ matrix.cuda-version }}
         run: |
           pip install torch==${{ matrix.torch-version}}+${{ matrix.cuda-version }} -f https://download.pytorch.org/whl/torch_stable.html
 
-      - name: Install main package
-        run: |
-          CUDA_HOME=/usr/local/cuda-11.1
-          LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
-          PATH=${CUDA_HOME}/bin:${PATH}
-          FORCE_CUDA=1 pip install -e .[test]
-
-      - name: Run test-suite
-        run: |
-          python setup.py test
-          ls -lah
-          ls torch_scatter -lah
-
-      - name: Build wheel
+      - name: Build main package
         run: |
+          pip install wheel
+          source .github/workflows/cuda/${{ matrix.cuda-version }}-${{ runner.os }}-env.sh
           python setup.py bdist_wheel --dist-dir=dist
+          ls -lah torch_scatter/
           ls -lah dist/

.github/workflows/cuda/cu111-Linux-env.sh

+#!/bin/bash
+
+CUDA_HOME=/usr/local/cuda-11.1
+LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
+PATH=${CUDA_HOME}/bin:${PATH}

.github/workflows/cu111-Linux.sh → .github/workflows/cuda/cu111-Linux.sh

@@ -11,9 +11,3 @@ sudo apt install cuda-nvcc-11-1 cuda-libraries-dev-11-1
 sudo apt clean
 
 rm -f https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda-repo-ubuntu2004-11-1-local_11.1.1-455.32.00-1_amd64.deb
-
-CUDA_HOME=/usr/local/cuda-11.1
-LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
-PATH=${CUDA_HOME}/bin:${PATH}
-
-nvcc --version