Start: Add option to not install wheels

Building from source is a case for many wheels, so add an option to skip wheel upgrades/installation if the user uses the start script. Signed-off-by: kingbri <[email protected]>
theroyallab · Dec 25, 2023 · 670ccac · 670ccac
1 parent 09ae71a
commit 670ccac
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/start.ps1 b/start.ps1
@@ -1,6 +1,7 @@
 # Arg parsing
 param(
     [switch]$ignore_upgrade = $false,
+    [switch]$nowheel = $false,
     [switch]$activate_venv = $false
 )
 
@@ -12,6 +13,12 @@ function GetRequirementsFile {
         exit
     }
 
+    # Install nowheel if specified
+    if ($nowheel) {
+        Write-Output "Not installing wheels due to user request."
+        return "requirements-nowheel"
+    }
+
     $CudaPath = $env:CUDA_PATH
     $CudaVersion = Split-Path $CudaPath -Leaf