chore(installer): overhaul install scripts — portable paths, TRT verification, pin alignment#10
Open
forkni wants to merge 1 commit intoSDTD_031_devfrom
Open
chore(installer): overhaul install scripts — portable paths, TRT verification, pin alignment#10forkni wants to merge 1 commit intoSDTD_031_devfrom
forkni wants to merge 1 commit intoSDTD_031_devfrom
Conversation
…fication, pin alignment - Replace hardcoded paths with %~dp0 in all batch scripts - Add prerequisite checks (Python 3.11, Git, optional cl.exe) - Align version pins across setup.py, installer.py, and all 3 TensorRT installers - Pin previously-unpinned python-osc to 1.10.2, fix timm install syntax - Fix protobuf phase7 downgrade below setup.py minimum (4.25.3→4.25.8) - Add TensorRT verification step (tensorrt, cuDNN, polygraphy checks) - Add set_env.bat for runtime GPU tuning (CUDA_MODULE_LOADING, L2 persist) - Update: Pillow>=12.2.0, onnxruntime-gpu==1.24.4, polygraphy==0.49.26, colored==2.3.2, pywin32==311, peft==0.18.1, timm==1.0.26 Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
2 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Summary
%~dp0inInstall_StreamDiffusion.bat,Install_TensorRT.bat, andStart_StreamDiffusion.bat. Added prerequisite checks (Python 3.11, Git, optional cl.exe warning).setup.py,sd_installer/installer.py,sd_installer/tensorrt.py,src/streamdiffusion/tools/install-tensorrt.py, andStreamDiffusionTD/install_tensorrt.pyto match verified installed state.TENSORRT_CHECKStoverifier.pyand wired intocli.pysoinstall-tensorrtautomatically verifies TRT, cuDNN, and polygraphy after install.set_env.bat: Runtime GPU tuning vars (PYTORCH_CUDA_ALLOC_CONF,CUDA_MODULE_LOADING=LAZY, L2 persist settings).python-oscto1.10.2, fixedtimminstall syntax bug, fixed protobuf phase7 downgrade belowsetup.pyminimum.Version pin changes
Files changed (9 files, installer-only)
Install_StreamDiffusion.bat— portable paths + prereq checksInstall_TensorRT.bat— portable paths + venv checkStart_StreamDiffusion.bat— portable paths + set_env.bat integrationset_env.bat— new: runtime GPU tuning environment variablessetup.py— 4 version pin updatesStreamDiffusion-installer— submodule: MANUAL_PINS, phase5 syntax, verifier TRT checksStreamDiffusionTD/install_tensorrt.py— centralized TENSORRT_PINS dict, aligned versionssrc/streamdiffusion/tools/install-tensorrt.py— polygraphy + pywin32 pin updatessrc/streamdiffusion/pip_utils.py— get_cuda_major helper for install-tensorrt.pyTest plan
py -3.11 -m sd_installer --base-folder "..." diagnose— all 14 checks passInstall_StreamDiffusion.batfrom a clean directory — portable paths resolve correctlyInstall_TensorRT.bat— TRT install + verification step succeedsimport tensorrt; import polygraphy; import onnxruntimeall succeed🤖 Generated with Claude Code