search

agimus-project / happypose

Toolbox and trackers for object pose-estimation. Based on the work CosyPose and MegaPose
https://agimus-project.github.io/happypose/
BSD 2-Clause "Simplified" License
23 stars 9 forks source link

documentation: gpu pypi url #157

Open nim65s opened 6 months ago

nim65s commented 6 months ago

README + https://agimus-project.github.io/happypose/ are showing

pip install .[pypi,cpu] --extra-index-url https://download.pytorch.org/whl/cpu

we have to explain that for gpu, there is another index url.

thanks @Kotochleb for raising this

nim65s commented 6 months ago

hum, actually, url for GPU depends on cuda version. And it might simply be pypi, as it is currently in pyproject.toml.

Let's wait for some pytorch / cuda / nvidia drivers updates, tests and decisions.