search

alex-petrenko / sample-factory

High throughput synchronous and asynchronous reinforcement learning
https://samplefactory.dev
MIT License
773 stars 106 forks source link

GPU Support for Apple M1 #264

Open visuallization opened 1 year ago

visuallization commented 1 year ago

Hey there,

Thanks for all the work you have put into this library! I was wondering if there will be support for GPU training on Apple M1? Apparently pytorch will be officially supporting GPU accelerated training on macs.

https://pytorch.org/blog/introducing-accelerated-pytorch-training-on-mac/ https://sebastianraschka.com/blog/2022/pytorch-m1-gpu.html

Cheers

alex-petrenko commented 1 year ago

I will soon get my hands on one of these laptops and it'd be great to add this support. I imagine it can be done with just a few changes in CUDA-specific code. Let's keep this feature open to discuss any development issues. Contributions are welcome!