Support for pytorch 1.13
benjats07 opened this issue ยท 3 comments
๐ Feature
Support for pytorch 1.13 and MPS
Motivation
I can't take advantage of MPS in macOS since is a recently added feature and lightning-flash didn't allow usage above torch 1.10
Alternatives
I already make config changes to force the installation with newer torch and torchvision versions, but seems that 'nms' isn't available on torchvision.ops.boxes and icevision request that.
lightning-flash didn't allow usage above torch 1.10
I don't this it is an issue with Flash, as you can see our dependencies... ๐ฐ
https://github.com/Lightning-AI/lightning-flash/blob/1e3395fe35014b997b97a59df9f8b24e0f272d1e/requirements.txt#L4
but it could come from other 3rd-party depenecy we rely on... ๐ค
Said so what is your use-case and how are you installing Flash? ๐ฆฆ
I want to train models using MPS acceleration on MacOS but this capability is only available on lastest version of Pytorch. Especifically I want to follow this https://lightning-flash.readthedocs.io/en/latest/reference/object_detection.html
You can try to force upgrade Torch; I think all shall be working fine...