Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Self-hosted Runner Edits #676

Open
wants to merge 14 commits into
base: master
Choose a base branch
from
2 changes: 1 addition & 1 deletion .github/scripts/execute_single_benchmark_self_hosted.sh
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
# Datasets are stored in /usr/local/gtsfm-data in order to reduce runtimes by not
# redownloading each (very large) dataset every CI run. Any new datasets must be
# downloaded and stored in /usr/local/gtsfm-data before running this action.
DATASET_PREFIX=/usr/local/gtsfm-data
DATASET_PREFIX=/home/akrishnan86/gtsfm/data

DATASET_NAME=$1
CONFIG_NAME=$2
Expand Down
3 changes: 3 additions & 0 deletions .github/workflows/benchmark-self-hosted.yml
Original file line number Diff line number Diff line change
Expand Up @@ -53,6 +53,9 @@ jobs:
# bash
# cache-environment: true
# post-cleanup: "all"
- name: Enter gtsfm directory + install it
run:
cd $GITHUB_WORKSPACE && pip install -e .
- name: Download third-party weights
run: |
./download_model_weights.sh
Expand Down
59 changes: 59 additions & 0 deletions environment_cluster.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,59 @@
name: gtsfm-v1
channels:
# This environment is identical to the default Linux environment except that
# we install a different PyTorch version that works on the cluster.
#
# for priority order, we prefer pytorch as the highest priority as it supplies
# latest stable packages for numerous deep learning based methods. conda-forge
# supplies higher versions of packages like opencv compared to the defaults
# channel.
- pytorch
- conda-forge
- nvidia
dependencies:
# python essentials
- python=3.8
- pip
# formatting and dev environment
- black
- coverage
- mypy
- pylint
- pytest
- flake8
- isort
# dask and related
- dask # same as dask[complete] pip distribution
- asyncssh
- python-graphviz
# core functionality and APIs
- matplotlib==3.4.2
- networkx
- numpy
- nodejs
- pandas
- pillow>=8.0.1
- scikit-learn
- scipy
- hydra-core
# 3rd party algorithms for different modules
- pytorch-cuda=11.8
- pytorch
- torchvision>=0.13.0
- kornia
# io
- h5py
- plotly=4.14.3
- tabulate
- simplejson
# testing
- parameterized
- pip:
- open3d
- opencv-python>=4.5.4.58
- pydegensac
- colour
- pycolmap>=0.1.0
- trimesh[easy]
- gtsam==4.2a8
- pydot
Loading