Skip to content

Commit

Permalink
build: use py 3.11 and cuda 12.4 by default
Browse files Browse the repository at this point in the history
  • Loading branch information
tazlin committed Jul 25, 2024
1 parent 0fc1954 commit 6f4498e
Show file tree
Hide file tree
Showing 7 changed files with 13 additions and 13 deletions.
2 changes: 1 addition & 1 deletion README_advanced.md
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@
### Get worker files and install dependencies
- `git clone https://github.com/Haidra-Org/horde-worker-reGen.git`
- `cd .\horde-worker-reGen\`
- `pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu121`
- `pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu124`

### Run worker
- Set your config now, copying `bridgeData_template.yaml` to `bridgeData.yaml`, being sure to set an API key and worker name at a minimum
Expand Down
2 changes: 1 addition & 1 deletion environment.linux.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -8,4 +8,4 @@ dependencies:
- nvidia::cuda-toolkit
- git
- pip
- python==3.10.12
- python==3.11.6
3 changes: 1 addition & 2 deletions environment.windows.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,6 @@ channels:
# These should only contain the minimal essentials to get the binaries going, everything else is managed in requirements.txt to keep it universal.
dependencies:
- nvidia::cuda-toolkit
- anaconda::vs2015_runtime
- git
- pip
- python==3.10.12
- python==3.11.6
9 changes: 5 additions & 4 deletions requirements.txt
Original file line number Diff line number Diff line change
@@ -1,10 +1,11 @@
horde_engine~=2.13.3
horde_sdk>=0.14.0
horde_model_reference>=0.8.1
horde_safety~=0.2.3

numpy==1.26.4
torch>=2.3.1

horde_sdk~=0.14.0
horde_safety~=0.2.3
horde_engine~=2.13.3
horde_model_reference>=0.8.1

python-dotenv
ruamel.yaml
Expand Down
2 changes: 1 addition & 1 deletion tox.ini
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ deps =
pytest-sugar
pytest-cov
requests
-r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu121
-r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu124
commands =
pytest tests {posargs} --cov

Expand Down
4 changes: 2 additions & 2 deletions update-runtime.cmd
Original file line number Diff line number Diff line change
Expand Up @@ -51,12 +51,12 @@ call "%MAMBA_ROOT_PREFIX%\condabin\micromamba.bat" activate windows

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 --extra-index-url https://download.pytorch.org/whl/cu121
python -s -m pip install horde_engine horde_model_reference --extra-index-url https://download.pytorch.org/whl/cu124
) else (
if defined scribe (
python -s -m pip install -r requirements-scribe.txt
) else (
python -s -m pip install -r requirements.txt -U --extra-index-url https://download.pytorch.org/whl/cu121
python -s -m pip install -r requirements.txt -U --extra-index-url https://download.pytorch.org/whl/cu124 -U
)
)
call deactivate
Expand Down
4 changes: 2 additions & 2 deletions update-runtime.sh
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ bin/micromamba create --no-shortcuts -r conda -n linux -f ${CONDA_ENVIRONMENT_FI

if [ "$hordelib" = true ]; then
bin/micromamba run -r conda -n linux python -s -m pip uninstall -y hordelib horde_engine horde_sdk horde_model_reference
bin/micromamba run -r conda -n linux python -s -m pip install horde_engine horde_model_reference --extra-index-url https://download.pytorch.org/whl/cu121
bin/micromamba run -r conda -n linux python -s -m pip install horde_engine horde_model_reference --extra-index-url https://download.pytorch.org/whl/cu124
else
bin/micromamba run -r conda -n linux python -s -m pip install -r requirements.txt -U --extra-index-url https://download.pytorch.org/whl/cu121
bin/micromamba run -r conda -n linux python -s -m pip install -r requirements.txt -U --extra-index-url https://download.pytorch.org/whl/cu124 -U
fi

0 comments on commit 6f4498e

Please sign in to comment.