Lightning-Universe/lightning-flash

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.

Borda commented

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

Borda commented

You can try to force upgrade Torch; I think all shall be working fine...