mirror of
https://github.com/RVC-Boss/GPT-SoVITS.git
synced 2025-10-16 14:08:38 +08:00
.
This commit is contained in:
parent
60c205ab66
commit
a6f942bd96
@ -228,7 +228,9 @@ switch ($Device) {
|
||||
Invoke-Pip torch torchaudio --index-url "https://download.pytorch.org/whl/cu128"
|
||||
Invoke-Conda cuda-nvcc=12.8
|
||||
Invoke-Pip psutil ninja packaging wheel "setuptools>=42"
|
||||
Write-Info "Installing Flash Attn..."
|
||||
Invoke-Pip flash-attn -i https://xxxxrt666.github.io/PIP-Index/ --no-build-isolation
|
||||
Write-Success "Flash Attn Installed"
|
||||
}
|
||||
"CU126" {
|
||||
$cudaLine = nvidia-smi | Select-String "CUDA Version"
|
||||
@ -241,7 +243,9 @@ switch ($Device) {
|
||||
Invoke-Pip torch torchaudio --index-url "https://download.pytorch.org/whl/cu126"
|
||||
Invoke-Conda cuda-nvcc=12.6
|
||||
Invoke-Pip psutil ninja packaging wheel "setuptools>=42"
|
||||
Write-Info "Installing Flash Attn..."
|
||||
Invoke-Pip flash-attn -i https://xxxxrt666.github.io/PIP-Index/ --no-build-isolation
|
||||
Write-Success "Flash Attn Installed"
|
||||
}
|
||||
"CPU" {
|
||||
Write-Info "Installing PyTorch For CPU..."
|
||||
|
Loading…
x
Reference in New Issue
Block a user