a-shell icon indicating copy to clipboard operation
a-shell copied to clipboard

PyTorch support

Open Drei-E3 opened this issue 2 years ago • 5 comments

is it possible to support PyTorch in the future?

Drei-E3 avatar Apr 05 '23 06:04 Drei-E3

Why you want to use an iOS device to do works with cuda? It can't be done by normal environments, requesting proper GPUs

HeavySnowJakarta avatar Apr 05 '23 15:04 HeavySnowJakarta

Why you want to use an iOS device to do works with cuda? It can't be done by normal environments, requesting proper GPUs

i bought the wrong macbook pro, it is extremely large and heavy. And i am also a fan of iPad. This project is cool and I wanna do something as well. When i realize that since ipad also has silicon chip so i wonder if ML Tasks could be processing on the iPad as well. And apple has announced that stable fusion support iPad GPU now, which however through swift.

And seriously iPadOS 16 is even stabler than the buggy macOS Ventura, which after updating my MBP cannot connect to the monitor with usb-c.

Drei-E3 avatar Apr 05 '23 17:04 Drei-E3

Why you want to use an iOS device to do works with cuda? It can't be done by normal environments, requesting proper GPUs

i bought the wrong macbook pro, it is extremely large and heavy. And i am also a fan of iPad. This project is cool and I wanna do something as well. When i realize that since ipad also has silicon chip so i wonder if ML Tasks could be processing on the iPad as well. And apple has announced that stable fusion support iPad GPU now, which however through swift.

And seriously iPadOS 16 is even stabler than the buggy macOS Ventura, which after updating my MBP cannot connect to the monitor with usb-c.

PyTorch for Apple silicon is based on Metal Performance Shaders, which is supported by iPadOS, so it is possible theoretically.

HeavySnowJakarta avatar Apr 06 '23 11:04 HeavySnowJakarta

PyTorch has already supported Apple MPS. It seems that iPad Pro with M2 support MPS?

CarlGao4 avatar Oct 06 '23 08:10 CarlGao4

Theoretically it can be ported as long as the environments needed are configured properly

HeavySnowJakarta avatar Oct 07 '23 10:10 HeavySnowJakarta