borglab / borglab/gtsfm

GPU computation improvements

Open
#293 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Jupyter Notebook
Stars
524
Forks
64
Avg merge
4h 1m
Merged PRs (30d)
3

Description

- Configure Dask to use all GPU devices that are available with $CUDA_VISIBLE_DEVICES, instead of using only GPU 0
- Figure out batched Pytorch - Dask interface to not reload models at every inference call

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.