Update Docker Scipts

This commit is contained in:
XXXXRT666 2026-02-17 02:46:10 +00:00
parent 0a8b006607
commit d5ed91deb5

View File

@ -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 "$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 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 "$HOME/conda/bin/conda" install cuda-nvcc=12.8 -y
elif [ "$CUDA_VERSION" = "12.6" ]; then 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 "$HOME/conda/bin/conda" install cuda-nvcc=12.6 -y
fi fi
export PATH="$HOME/conda/bin:$PATH" 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]" --python "$(which python)"
"$HOME/conda/bin/uv" pip install flash-attn -i https://xxxxrt666.github.io/PIP-Index/ --no-build-isolation
"$HOME/conda/bin/uv" cache clean "$HOME/conda/bin/uv" cache clean
rm $LOG_PATH rm $LOG_PATH