diff --git a/start.ps1 b/start.ps1 index 3919f5e..6642db9 100644 --- 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