update-runtime-directml.cmd

@echo off
cd /d "%~dp0"

SET MAMBA_ROOT_PREFIX=%~dp0conda
echo %MAMBA_ROOT_PREFIX%

if exist "%MAMBA_ROOT_PREFIX%\condabin\micromamba.bat" (
    echo Deleting micromamba.exe as its out of date
    del micromamba.exe
    if errorlevel 1 (
        echo Error: Failed to delete micromamba.exe. Please delete it manually.
        exit /b 1
    )
    echo Deleting the conda directory as its out of date
    rmdir /s /q conda
    if errorlevel 1 (
        echo Error: Failed to delete the conda directory. Please delete it manually.
        exit /b 1
    )
)

:Check if micromamba is already installed
if exist micromamba.exe goto Isolation
  curl.exe -L -o micromamba.exe https://github.com/mamba-org/micromamba-releases/releases/latest/download/micromamba-win-64


:Isolation
SET CONDA_SHLVL=
SET PYTHONNOUSERSITE=1
SET PYTHONPATH=
echo %MAMBA_ROOT_PREFIX%



setlocal EnableDelayedExpansion
for %%a in (%*) do (
    if /I "%%a"=="--hordelib" (
        set hordelib=true
    ) else (
        set hordelib=
    )
    if /I "%%a"=="--scribe" (
        set scribe=true
    ) else (
        set scribe=
    )
)
endlocal

if defined scribe (
  SET CONDA_ENVIRONMENT_FILE=environment_scribe.yaml

) else (
  SET CONDA_ENVIRONMENT_FILE=environment.rocm.yaml
)

Reg add "HKLM\SYSTEM\CurrentControlSet\Control\FileSystem" /v "LongPathsEnabled" /t REG_DWORD /d "1" /f 2>nul
:We do this twice the first time to workaround a conda bug where pip is not installed correctly the first time - Henk
IF EXIST CONDA GOTO WORKAROUND_END
.\micromamba.exe create --no-shortcuts -r conda -n windows -f %CONDA_ENVIRONMENT_FILE% -y
:WORKAROUND_END
.\micromamba.exe create --no-shortcuts -r conda -n windows -f %CONDA_ENVIRONMENT_FILE% -y

REM Check if hordelib argument is defined

micromamba.exe shell hook -s cmd.exe %MAMBA_ROOT_PREFIX% -v
call "%MAMBA_ROOT_PREFIX%\condabin\mamba_hook.bat"
call "%MAMBA_ROOT_PREFIX%\condabin\mamba.bat" activate windows

python -s -m pip install torch-directml torchvision==0.19.1

if defined hordelib (
  python -s -m pip uninstall -y hordelib horde_engine horde_model_reference
  python -s -m pip install horde_engine horde_model_reference
) else (
  if defined scribe (
    python -s -m pip install -r requirements-scribe.txt
  ) else (
    python -s -m pip install -r requirements.directml.txt
  )
)
call deactivate

echo If there are no errors above everything should be correctly installed (If not, try deleting the folder /conda/envs/ and try again).

pause