mirror of
https://github.com/vegu-ai/talemate.git
synced 2025-09-01 18:09:08 +00:00
separate cuda install
This commit is contained in:
parent
6a62b4e67e
commit
26c21193b3
2 changed files with 18 additions and 0 deletions
8
install-cuda.bat
Normal file
8
install-cuda.bat
Normal file
|
@ -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
|
10
install-cuda.sh
Executable file
10
install-cuda.sh
Executable file
|
@ -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
|
Loading…
Add table
Reference in a new issue