Use pytorch 2.7 (with cuda 12.8) on new installations with NVIDIA gpus

This commit is contained in:
cmdr2
2025-04-24 15:50:55 +05:30
parent a33737b991
commit e0a4d43087
3 changed files with 7 additions and 7 deletions

View File

@ -47,7 +47,7 @@ if [ -e "src" ]; then mv src src-old; fi
if [ -e "ldm" ]; then mv ldm ldm-old; fi
# this is outside check_modules.py to ensure that the required version of torchruntime is present
python -m pip install -q "torchruntime>=1.15.1"
python -m pip install -q "torchruntime>=1.16.2"
cd ..
# Download the required packages