31 lines
565 B
Batchfile
31 lines
565 B
Batchfile
@echo off
|
|
|
|
set MODULE_NAME=pytorch
|
|
|
|
IF NOT EXIST "setup.py" IF NOT EXIST "%MODULE_NAME%" (
|
|
call internal\clone.bat
|
|
cd %~dp0
|
|
) ELSE (
|
|
call internal\clean.bat
|
|
)
|
|
|
|
IF ERRORLEVEL 1 goto :eof
|
|
|
|
call internal\check_deps.bat
|
|
IF ERRORLEVEL 1 goto :eof
|
|
|
|
REM Check for optional components
|
|
|
|
echo Disabling CUDA
|
|
set USE_CUDA=0
|
|
|
|
call internal\check_opts.bat
|
|
IF ERRORLEVEL 1 goto :eof
|
|
|
|
if exist "%NIGHTLIES_PYTORCH_ROOT%" cd %NIGHTLIES_PYTORCH_ROOT%\..
|
|
call %~dp0\internal\copy_cpu.bat
|
|
IF ERRORLEVEL 1 goto :eof
|
|
|
|
call %~dp0\internal\setup.bat
|
|
IF ERRORLEVEL 1 goto :eof
|