diff --git a/install.ps1 b/install.ps1 index 4646a60e..2e93c1ef 100644 --- a/install.ps1 +++ b/install.ps1 @@ -221,7 +221,7 @@ switch ($Device) { $cudaLine = nvidia-smi | Select-String "CUDA Version" $version = ($cudaLine -split "CUDA Version:")[1].Trim() Write-Info "Maximum CUDA Version Supported By Current Driver: $version" - if ([version](nvidia-smi | Select-String "CUDA Version" | ForEach-Object { ($_ -split "CUDA Version:")[1].Trim() }) -ge [version]"12.8") { + if ([version](nvidia-smi | Select-String "CUDA Version" | ForEach-Object { ($_ -split "CUDA Version:")[1].Trim() }) -ge [version]"12.0") { Write-Warning "CUDA 12.8 Is Not Supported By Current Driver" } Write-Info "Installing PyTorch For CUDA 12.8..." @@ -236,7 +236,7 @@ switch ($Device) { $cudaLine = nvidia-smi | Select-String "CUDA Version" $version = ($cudaLine -split "CUDA Version:")[1].Trim() Write-Info "Maximum CUDA Version Supported By Current Driver: $version" - if ([version](nvidia-smi | Select-String "CUDA Version" | ForEach-Object { ($_ -split "CUDA Version:")[1].Trim() }) -ge [version]"12.8") { + if ([version](nvidia-smi | Select-String "CUDA Version" | ForEach-Object { ($_ -split "CUDA Version:")[1].Trim() }) -ge [version]"12.0") { Write-Warning "CUDA 12.6 Is Not Supported By Current Driver" } Write-Info "Installing PyTorch For CUDA 12.6..." diff --git a/install.sh b/install.sh index e7d298af..83d2b778 100644 --- a/install.sh +++ b/install.sh @@ -330,15 +330,15 @@ if [ "$USE_CUDA" = true ] && [ "$WORKFLOW" = false ]; then CUDAVERSION=$(nvidia-smi | grep "CUDA Version" | sed -E 's/.*CUDA Version: ([0-9]+\.[0-9]+).*/\1/') echo -e "${INFO}Maximum CUDA Version Supported By Current Driver: $CUDAVERSION" if [ "$CUDA" = 128 ]; then - if awk "BEGIN {exit !($CUDAVERSION < 12.8)}"; then - echo -r "${WARNING}CUDA 12.8 Is Not Supported By Current Driver" + if awk "BEGIN {exit !($CUDAVERSION < 12.0)}"; then + echo -e "${WARNING}CUDA 12.8 Is Not Supported By Current Driver" fi echo -e "${INFO}Installing PyTorch For CUDA 12.8..." run_pip_quiet torch torchao --index-url "https://download.pytorch.org/whl/cu128" run_conda_quiet cuda-nvcc=12.8 elif [ "$CUDA" = 126 ]; then - if awk "BEGIN {exit !($CUDAVERSION < 12.6)}"; then - echo -r "${WARNING}CUDA 12.6 Is Not Supported By Current Driver" + if awk "BEGIN {exit !($CUDAVERSION < 12.0)}"; then + echo -e "${WARNING}CUDA 12.6 Is Not Supported By Current Driver" fi echo -e "${INFO}Installing PyTorch For CUDA 12.6..." run_pip_quiet torch torchao --index-url "https://download.pytorch.org/whl/cu126"