Training PyTorch on M1 Macs

deepjavalibrary / djl

An Engine-Agnostic Deep Learning Framework in Java

https://djl.ai

Apache License 2.0

4.07k stars 650 forks source link

Training PyTorch on M1 Macs #2103

Closed carlosuc3m closed 1 year ago

carlosuc3m commented 1 year ago

Hello again, with the new possibility of training PyTorch models in Macs with the M1 chip I was wondering if that feature will ever be added to DJL. I have already seen dependencies with support for osx-arch: https://mvnrepository.com/artifact/ai.djl.pytorch/pytorch-jni

Regards, Carlos

zachgk commented 1 year ago

We do have support for M1 PyTorch so you should be able to train on them now

github-actions[bot] commented 1 year ago

This issue has been automatically closed because there has been no response to our request for more information from the original author. With only the information that is currently in the issue, we don't have enough information to take action. Please reach out if you have or find the answers we need so that we can investigate further.