mirror of
https://github.com/RVC-Boss/GPT-SoVITS.git
synced 2026-06-03 20:40:30 +08:00
Update Docker Scipts
This commit is contained in:
parent
0a8b006607
commit
d5ed91deb5
@ -60,17 +60,16 @@ source "$HOME/.bashrc"
|
||||
"$HOME/conda/bin/conda" install gcc=11 gxx ffmpeg uv cmake make unzip $SYSROOT_PKG "libstdcxx-ng>=11" -y
|
||||
|
||||
if [ "$CUDA_VERSION" = "12.8" ]; then
|
||||
"$HOME/conda/bin/uv" pip install torch torchcodec --no-cache-dir --index-url https://download.pytorch.org/whl/cu128
|
||||
"$HOME/conda/bin/uv" pip install ".[cu128]" --no-cache-dir --python "$(which python)"
|
||||
"$HOME/conda/bin/conda" install cuda-nvcc=12.8 -y
|
||||
elif [ "$CUDA_VERSION" = "12.6" ]; then
|
||||
"$HOME/conda/bin/uv" pip install torch torchcodec --no-cache-dir --index-url https://download.pytorch.org/whl/cu126
|
||||
"$HOME/conda/bin/uv" pip install ".[cu126]" --no-cache-dir --python "$(which python)"
|
||||
"$HOME/conda/bin/conda" install cuda-nvcc=12.6 -y
|
||||
fi
|
||||
|
||||
export PATH="$HOME/conda/bin:$PATH"
|
||||
|
||||
"$HOME/conda/bin/uv" pip install psutil ninja packaging wheel "setuptools>=42" einops
|
||||
"$HOME/conda/bin/uv" pip install flash-attn -i https://xxxxrt666.github.io/PIP-Index/ --no-build-isolation
|
||||
"$HOME/conda/bin/uv" pip install ".[flash-attn]" --python "$(which python)"
|
||||
"$HOME/conda/bin/uv" cache clean
|
||||
|
||||
rm $LOG_PATH
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user