MPS for segmentation - Githubissues

koito19960406 / ZenSVI

This package is a one-stop solution for downloading, cleaning, analyzing street view imagery

https://zensvi.readthedocs.io/en/latest/

Creative Commons Attribution Share Alike 4.0 International

19 stars 2 forks source link

Closed koito19960406 closed 10 months ago

koito19960406 commented 10 months ago

MPS is now available in segmentation. make sure to:

Use torch => 2.1.0 (pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu)
Set environment variable: conda env config vars set PYTORCH_ENABLE_MPS_FALLBACK=1