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-Pip torch torchaudio --index-url "https://download.pytorch.org/whl/cu128"
|
||||||
Invoke-Conda cuda-nvcc=12.8
|
Invoke-Conda cuda-nvcc=12.8
|
||||||
Invoke-Pip psutil ninja packaging wheel "setuptools>=42"
|
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
|
Invoke-Pip flash-attn -i https://xxxxrt666.github.io/PIP-Index/ --no-build-isolation
|
||||||
|
Write-Success "Flash Attn Installed"
|
||||||
}
|
}
|
||||||
"CU126" {
|
"CU126" {
|
||||||
$cudaLine = nvidia-smi | Select-String "CUDA Version"
|
$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-Pip torch torchaudio --index-url "https://download.pytorch.org/whl/cu126"
|
||||||
Invoke-Conda cuda-nvcc=12.6
|
Invoke-Conda cuda-nvcc=12.6
|
||||||
Invoke-Pip psutil ninja packaging wheel "setuptools>=42"
|
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
|
Invoke-Pip flash-attn -i https://xxxxrt666.github.io/PIP-Index/ --no-build-isolation
|
||||||
|
Write-Success "Flash Attn Installed"
|
||||||
}
|
}
|
||||||
"CPU" {
|
"CPU" {
|
||||||
Write-Info "Installing PyTorch For CPU..."
|
Write-Info "Installing PyTorch For CPU..."
|
||||||
|
Loading…
x
Reference in New Issue
Block a user