diff --git a/install-cuda.bat b/install-cuda.bat new file mode 100644 index 00000000..d49221b8 --- /dev/null +++ b/install-cuda.bat @@ -0,0 +1,8 @@ +REM activate the virtual environment +call talemate_env\Scripts\activate + +REM uninstall torch and torchaudio +python -m pip uninstall torch torchaudio -y + +REM install torch and torchaudio +python -m pip install torch~=2.7.0 torchaudio~=2.7.0 --index-url https://download.pytorch.org/whl/cu128 \ No newline at end of file diff --git a/install-cuda.sh b/install-cuda.sh new file mode 100755 index 00000000..59c904f0 --- /dev/null +++ b/install-cuda.sh @@ -0,0 +1,10 @@ +#!/bin/bash + +# activate the virtual environment +source talemate_env/bin/activate + +# uninstall torch and torchaudio +python -m pip uninstall torch torchaudio -y + +# install torch and torchaudio +python -m pip install torch~=2.7.0 torchaudio~=2.7.0 --index-url https://download.pytorch.org/whl/cu128 \ No newline at end of file