Fix for blackwell GPUs on new installations

This commit is contained in:
cmdr2 2025-03-29 14:34:17 +05:30
parent 1b3d048bf2
commit 45c6865b2d
2 changed files with 4 additions and 2 deletions

View File

@ -70,7 +70,8 @@ echo PYTHONPATH=%PYTHONPATH%
call where python call where python
call python --version call python --version
call python -m pip install -q torchruntime @rem this is outside check_modules.py to ensure that the required version of torchruntime is present
call python -m pip install -q torchruntime>=1.15.1
call python scripts\check_modules.py --launch-uvicorn call python scripts\check_modules.py --launch-uvicorn
pause pause

View File

@ -46,7 +46,8 @@ fi
if [ -e "src" ]; then mv src src-old; fi if [ -e "src" ]; then mv src src-old; fi
if [ -e "ldm" ]; then mv ldm ldm-old; fi if [ -e "ldm" ]; then mv ldm ldm-old; fi
python -m pip install -q torchruntime # 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
cd .. cd ..
# Download the required packages # Download the required packages