Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- @echo off
- if not defined PYTHON (set PYTHON=python)
- if not defined GIT (set GIT=git)
- if not defined COMMANDLINE_ARGS (set COMMANDLINE_ARGS=%*)
- if not defined VENV_DIR (set VENV_DIR=venv)
- if not defined TORCH_COMMAND (set TORCH_COMMAND=pip install torch==1.12.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113)
- if not defined REQS_FILE (set REQS_FILE=requirements_versions.txt)
- mkdir tmp 2>NUL
- %PYTHON% -c "" >tmp/stdout.txt 2>tmp/stderr.txt
- if %ERRORLEVEL% == 0 goto :setup_venv
- echo Couldn't launch python
- goto :show_stdout_stderr
- :setup_venv
- if [%VENV_DIR%] == [-] goto :skip_venv
- dir %VENV_DIR%\Scripts\Python.exe >tmp/stdout.txt 2>tmp/stderr.txt
- if %ERRORLEVEL% == 0 goto :activate_venv
- for /f "delims=" %%i in ('CALL %PYTHON% -c "import sys; print(sys.executable)"') do set PYTHON_FULLNAME="%%i"
- echo Creating venv in directory %VENV_DIR% using python %PYTHON_FULLNAME%
- %PYTHON_FULLNAME% -m venv %VENV_DIR% >tmp/stdout.txt 2>tmp/stderr.txt
- if %ERRORLEVEL% == 0 goto :activate_venv
- echo Unable to create venv in directory %VENV_DIR%
- goto :show_stdout_stderr
- :activate_venv
- set PYTHON="%~dp0%VENV_DIR%\Scripts\Python.exe"
- %PYTHON% --version
- echo venv %PYTHON%
- goto :install_torch
- :skip_venv
- echo Skip
- %PYTHON% --version
- :install_torch
- %PYTHON% -c "import torch" >tmp/stdout.txt 2>tmp/stderr.txt
- if %ERRORLEVEL% == 0 goto :check_gpu
- echo Installing torch...
- %PYTHON% -m %TORCH_COMMAND% >tmp/stdout.txt 2>tmp/stderr.txt
- if %ERRORLEVEL% == 0 goto :check_gpu
- echo Failed to install torch
- goto :show_stdout_stderr
- :check_gpu
- %PYTHON% -c "import torch; assert torch.cuda.is_available(), 'CUDA is not available'" >tmp/stdout.txt 2>tmp/stderr.txt
- if %ERRORLEVEL% == 0 goto :launch
- echo Torch is not able to use GPU
- goto :show_stdout_stderr
- :launch
- echo Launching prune.py...
- %PYTHON% prune.py
- pause
- exit /b
- :show_stdout_stderr
- echo.
- echo exit code: %errorlevel%
- for /f %%i in ("tmp\stdout.txt") do set size=%%~zi
- if %size% equ 0 goto :show_stderr
- echo.
- echo stdout:
- type tmp\stdout.txt
- :show_stderr
- for /f %%i in ("tmp\stderr.txt") do set size=%%~zi
- if %size% equ 0 goto :show_stderr
- echo.
- echo stderr:
- type tmp\stderr.txt
- :endofscript
- echo.
- echo Launch unsuccessful. Exiting.
- pause
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement