This commit is contained in:
XXXXRT666 2025-04-30 23:27:21 +01:00
parent b6d7e971a9
commit 0ac28b42ae
2 changed files with 6 additions and 2 deletions

View File

@ -41,7 +41,7 @@ rm anaconda.sh
rm $LOG_PATH
sudo rm -rf "$HOME/anaconda3/pkgs"
rm -rf "$HOME/anaconda3/pkgs"
mkdir "$HOME/anaconda3/pkgs"

View File

@ -52,6 +52,10 @@ conda config --add channels conda-forge
conda update --all -y
rm -rf "$HOME/anaconda3/pkgs"
mkdir "$HOME/anaconda3/pkgs"
if [ "$CUDA_VERSION" = 128 ]; then
pip install torch torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/cu128
elif [ "$CUDA_VERSION" = 124 ]; then
@ -72,7 +76,7 @@ pip show torch
rm -rf /tmp/* /var/tmp/*
sudo rm -rf "$HOME/anaconda3/pkgs"
rm -rf "$HOME/anaconda3/pkgs"
mkdir "$HOME/anaconda3/pkgs"