Faster build, fix "no space left".

.github/workflows/build.yaml

@@ -47,45 +47,63 @@ jobs:
         shell: pwsh
     env:
         CUDA_VERSION: ${{ matrix.cuda }}
-
+    
     steps:
+      - name: Free Disk Space
+        uses: jlumbroso/free-disk-space@v1.3.0
+        if: runner.os == 'Linux'
+        with:
+          tool-cache: false
+          android: true
+          dotnet: true
+          haskell: true
+          large-packages: false
+          docker-images: true
+          swap-storage: false
+        
       - uses: actions/checkout@v3
 
       - uses: actions/setup-python@v3
         with:
           python-version: ${{ matrix.pyver }}
 
-      - name: Setup Miniconda
+      - name: Setup Mamba
         uses: conda-incubator/setup-miniconda@v2.2.0
         with:
           activate-environment: "build"
           python-version: ${{ matrix.pyver }}
-          mamba-version: "*"
-          use-mamba: false
-          channels: conda-forge,defaults
-          channel-priority: true
+          miniforge-variant: Mambaforge
+          miniforge-version: latest
+          use-mamba: true
           add-pip-as-python-dependency: true
           auto-activate-base: false
-
+          
       - name: Install Dependencies
         run: |
-          conda install cuda-toolkit -c "nvidia/label/cuda-${env:CUDA_VERSION}.0"
-          conda install pytorch "pytorch-cuda=${env:CUDA_VERSION}" -c pytorch -c nvidia
-          python -m pip install --upgrade build setuptools wheel ninja
+          # Install CUDA toolkit
+          mamba install -y 'cuda' -c "nvidia/label/cuda-${env:CUDA_VERSION}.0"
 
-          # Environment variables
-          Add-Content $env:GITHUB_ENV "CUDA_PATH=$env:CONDA_PREFIX"
-          Add-Content $env:GITHUB_ENV "CUDA_HOME=$env:CONDA_PREFIX"
-          if ($IsLinux) {$env:LD_LIBRARY_PATH = $env:CONDA_PREFIX + '/lib:' + $env:LD_LIBRARY_PATH}
+          # Env variables
+          $env:CUDA_PATH = $env:CONDA_PREFIX
+          $env:CUDA_HOME = $env:CONDA_PREFIX
+          
+          # Install torch
+          $cudaVersionPytorch = $env:CUDA_VERSION.Replace('.', '')
+          python -m pip install --upgrade --no-cache-dir torch==2.0.1+cu$cudaVersionPytorch --index-url https://download.pytorch.org/whl/cu$cudaVersionPytorch
+          python -m pip install build setuptools wheel ninja
 
           # Print version information
           python --version
           python -c "import torch; print('PyTorch:', torch.__version__)"
           python -c "import torch; print('CUDA:', torch.version.cuda)"
+          python -c "import os; print('CUDA_HOME:', os.getenv('CUDA_HOME', None))"
           python -c "from torch.utils import cpp_extension; print (cpp_extension.CUDA_HOME)"
 
       - name: Build Wheel
         run: |
+          $env:CUDA_PATH = $env:CONDA_PREFIX
+          $env:CUDA_HOME = $env:CONDA_PREFIX
+
           python setup.py sdist bdist_wheel
       
       - name: Upload Assets