added scripts to simplify windows installation

This commit is contained in:
Li 2023-08-14 01:27:10 -07:00
parent 1975754162
commit 19bd36147a
3 changed files with 84 additions and 0 deletions

21
first-time-runner.bat Normal file
View File

@ -0,0 +1,21 @@
@echo off
set "filePath=%cd%\webui-user.bat"
(
echo @echo off
echo.
echo set GIT=
echo set VENV_DIR=
echo set COMMANDLINE_ARGS=--skip-torch-cuda-test --precision full --no-half
echo set PYTORCH_TRACING_MODE=TORCHFX
echo.
echo call webui.bat
) > %filepath%
call webui-user.bat
pause

12
torch-install.bat Normal file
View File

@ -0,0 +1,12 @@
@echo off
start /wait cmd /k "%cd%\venv\Scripts\activate && pip install %cd%\torch-2.1.0.dev20230713+cpu-cp310-cp310-win_amd64.whl && exit"
echo torch 2.1.0 dev installation completed.
powershell -executionpolicy bypass .\torch-install.ps1
echo eval_frame.py modification completed. press any key to exit
pause

51
torch-install.ps1 Normal file
View File

@ -0,0 +1,51 @@
$scriptDirectory = $PSScriptRoot
Set-Location $scriptDirectory
## modify webui-user.bat
$filePath = $pwd.Path + "\webui-user.bat"
$newContent = @"
@echo off
set PYTHON=
set GIT=
set VENV_DIR=
set COMMANDLINE_ARGS=--skip-torch-cuda-test --precision full --no-half --skip-prepare-environment
set PYTORCH_TRACING_MODE=TORCHFX
call webui.bat
"@
$newContent | Set-Content -Path $filePath
### modify eval_frame
$eval_filePath = $pwd.Path + "\venv\Lib\site-packages\torch\_dynamo\eval_frame.py"
#comment out the two lines to test torch.compile on windows
$replacements = @{
" if sys.platform == `"win32`":" = "# if sys.platform == `"win32`":"
" raise RuntimeError(`"Windows not yet supported for torch.compile`")" = "# raise RuntimeError(`"Windows not yet supported for torch.compile`")"
}
$lines = Get-Content -Path $eval_filePath
foreach ($search_Text in $replacements.Keys){
$replaceText = $replacements[$search_text]
$lines = $lines.Replace($search_Text , $replaceText)
}
#write content back to file
$lines | Set-Content -Path $eval_filePath