diff --git a/scripts/linux/linux_install.sh b/scripts/linux/linux_install.sh index 4b5f62ff..d76b2a47 100644 --- a/scripts/linux/linux_install.sh +++ b/scripts/linux/linux_install.sh @@ -49,7 +49,7 @@ gpuchoice="${gpuchoice:0:1}" if [[ "${gpuchoice^^}" == "A" ]]; then PACKAGES_TO_INSTALL="python=3.10 cuda-toolkit ninja git gcc" - CHANNEL="-c nvidia/label/cuda-11.8.0 -c nvidia -c conda-forge" + CHANNEL="-c nvidia/label/cuda-12.1.1 -c nvidia -c conda-forge" elif [[ "${gpuchoice^^}" == "B" ]]; then PACKAGES_TO_INSTALL="python=3.10 rocm-comgr rocm-smi ninja git gcc" CHANNEL=" -c conda-forge" @@ -99,7 +99,7 @@ if [ ! -d "$INSTALL_ENV_DIR" ]; then echo "Packages to install: $PACKAGES_TO_INSTALL" conda create -y -k -p "$INSTALL_ENV_DIR" $CHANNEL $PACKAGES_TO_INSTALL || ( echo && echo "Conda environment creation failed." && exit 1 ) if [[ "${gpuchoice^^}" == "A" ]]; then - conda run --live-stream -p "$INSTALL_ENV_DIR" python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 || ( echo && echo "Pytorch installation failed." && exit 1 ) + conda run --live-stream -p "$INSTALL_ENV_DIR" python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 || ( echo && echo "Pytorch installation failed." && exit 1 ) elif [[ "${gpuchoice^^}" == "B" ]]; then conda run --live-stream -p "$INSTALL_ENV_DIR" python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm5.4.2 || ( echo && echo "Pytorch installation failed." && exit 1 ) elif [[ "${gpuchoice^^}" == "C" ]]; then diff --git a/scripts/macos/macos_install.sh b/scripts/macos/macos_install.sh index f35f8056..2faae30e 100644 --- a/scripts/macos/macos_install.sh +++ b/scripts/macos/macos_install.sh @@ -44,7 +44,7 @@ gpuchoice="${gpuchoice:0:1}" uppercase_gpuchoice=$(echo "$gpuchoice" | tr '[:lower:]' '[:upper:]') if [[ "$uppercase_gpuchoice" == "A" ]]; then PACKAGES_TO_INSTALL="python=3.10 cuda-toolkit ninja git" - CHANNEL="-c pytorch -c conda-forge" + CHANNEL="-c nvidia/label/cuda-12.1.1 -c nvidia -c conda-forge" elif [[ "$uppercase_gpuchoice" == "B" ]]; then PACKAGES_TO_INSTALL="python=3.10 ninja git" CHANNEL="-c conda-forge" diff --git a/scripts/windows/win_install.bat b/scripts/windows/win_install.bat index 15892a7e..a49b2be2 100644 --- a/scripts/windows/win_install.bat +++ b/scripts/windows/win_install.bat @@ -45,7 +45,7 @@ set gpuchoice=%gpuchoice:~0,1% if /I "%gpuchoice%" == "A" ( set "PACKAGES_TO_INSTALL=python=3.10 cuda-toolkit ninja git" - set "CHANNEL=-c nvidia/label/cuda-11.8.0 -c nvidia -c conda-forge" + set "CHANNEL=-c nvidia/label/cuda-12.1.1 -c nvidia -c conda-forge" ) else if /I "%gpuchoice%" == "B" ( set "PACKAGES_TO_INSTALL=python=3.10 rocm-comgr rocm-smi ninja git" set "CHANNEL=-c conda-forge" @@ -107,7 +107,7 @@ if /I "%gpuchoice%" == "B" ( if not exist "%INSTALL_ENV_DIR%" ( echo Packages to install: %PACKAGES_TO_INSTALL% call conda create --no-shortcuts -y -k -p "%INSTALL_ENV_DIR%" %CHANNEL% %PACKAGES_TO_INSTALL% || ( echo. && echo Conda environment creation failed. && goto end ) - if /I "%gpuchoice%" == "A" call conda run --live-stream -p "%INSTALL_ENV_DIR%" python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118|| ( echo. && echo Pytorch installation failed.&& goto end ) + if /I "%gpuchoice%" == "A" call conda run --live-stream -p "%INSTALL_ENV_DIR%" python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121|| ( echo. && echo Pytorch installation failed.&& goto end ) if /I "%gpuchoice%" == "B" call conda run --live-stream -p "%INSTALL_ENV_DIR%" python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm5.4.2|| ( echo. && echo Pytorch installation failed.&& goto end ) if /I "%gpuchoice%" == "C" call conda run --live-stream -p "%INSTALL_ENV_DIR%" python -m pip install torch torchvision torchaudio|| ( echo. && echo Pytorch installation failed.&& goto end ) )