[skip ci] build conda

.github/workflows/building-conda.yml

@@ -35,14 +35,24 @@ jobs:
 
       - name: Install Conda packages
         run: |
-          conda install conda-build
-          conda install conda-verify
+          conda install conda-build conda-verify
 
       - name: Install CUDA ${{ matrix.cuda-version }}
         if: ${{ matrix.cuda-version != 'cpu' }}
         run: |
           bash .github/workflows/cuda/${{ matrix.cuda-version }}-${{ runner.os }}.sh
 
-      - name: Build Conda package
+      - name: Build Conda package for CPU
+        if: ${{ matrix.cuda-version == 'cpu' }}
         run: |
-          bash ./conda/pytorch-spline-conv/build_conda.sh ${{ matrix.torch-version }} ${{ matrix.cuda-version }}
+          FORCE_ONLY_CPU=1 source ./conda/pytorch-spline-conv/build_conda.sh ${{ matrix.torch-version }} ${{ matrix.cuda-version }}
+        shell:
+          bash
+
+      - name: Build Conda package for GPU
+        if: ${{ matrix.cuda-version != 'cpu' }}
+        run: |
+          source .github/workflows/cuda/${{ matrix.cuda-version }}-${{ runner.os }}-env.sh
+          FORCE_CUDA=1 source ./conda/pytorch-spline-conv/build_conda.sh ${{ matrix.torch-version }} ${{ matrix.cuda-version }}
+        shell:
+          bash

conda/pytorch-spline-conv/build_conda.sh

@@ -6,10 +6,8 @@ export CUDA_VERSION=$2
 export CONDA_PYTORCH_CONSTRAINT="pytorch==${TORCH_VERSION%.*}.*"
 
 if [ "${CUDA_VERSION}" = "cpu" ]; then
-  export FORCE_ONLY_CPU=1
   export CONDA_CUDATOOLKIT_CONSTRAINT="cpuonly"
 else
-  export FORCE_CUDA=1
   case $CUDA_VERSION in
     cu111)
       export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit==11.1.*"

conda/pytorch-spline-conv/meta.yaml

@@ -23,6 +23,7 @@ build:
   string: py{{py}}_torch_{{ environ['TORCH_VERSION'] }}_{{ environ['CUDA_VERSION'] }}
   script: python setup.py install
   script_env:
+    - CUDA_HOME
     - FORCE_CUDA
     - FORCE_ONLY_CPU