This commit is contained in:
XXXXRT666 2025-05-01 03:05:50 +01:00
parent 1c780def2e
commit c14575ffec
3 changed files with 8 additions and 2 deletions

View File

@ -27,13 +27,13 @@ jobs:
include: include:
- cuda_version: 12.4 - cuda_version: 12.4
lite: true lite: true
tag_prefix: lite-cu124 tag_prefix: cu124-lite
- cuda_version: 12.4 - cuda_version: 12.4
lite: false lite: false
tag_prefix: cu124 tag_prefix: cu124
- cuda_version: 12.8 - cuda_version: 12.8
lite: true lite: true
tag_prefix: lite-cu128 tag_prefix: cu128-lite
- cuda_version: 12.8 - cuda_version: 12.8
lite: false lite: false
tag_prefix: cu128 tag_prefix: cu128

View File

@ -45,6 +45,8 @@ source "$HOME/anaconda3/etc/profile.d/conda.sh"
"$HOME/anaconda3/bin/conda" update -q --all -y 1>/dev/null "$HOME/anaconda3/bin/conda" update -q --all -y 1>/dev/null
"$HOME/anaconda3/bin/conda" install python=3.11 -q -y
rm $LOG_PATH rm $LOG_PATH
rm -rf "$HOME/anaconda3/pkgs" rm -rf "$HOME/anaconda3/pkgs"

View File

@ -48,9 +48,13 @@ fi
source "$HOME/anaconda3/etc/profile.d/conda.sh" source "$HOME/anaconda3/etc/profile.d/conda.sh"
echo "CUDA_VERSION: $CUDA_VERSION"
if [ "$CUDA_VERSION" = 128 ]; then if [ "$CUDA_VERSION" = 128 ]; then
echo 1111111
pip install torch torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/cu128 pip install torch torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/cu128
elif [ "$CUDA_VERSION" = 124 ]; then elif [ "$CUDA_VERSION" = 124 ]; then
echo 2222222
pip install torch==2.5.1 torchaudio==2.5.1 --no-cache-dir --index-url https://download.pytorch.org/whl/cu124 pip install torch==2.5.1 torchaudio==2.5.1 --no-cache-dir --index-url https://download.pytorch.org/whl/cu124
fi fi