Enable Apple Silicon GPU Acceleration for d2l

d2l-ai / d2l-en

Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.

https://D2L.ai

Other

22.45k stars 4.19k forks source link

Enable Apple Silicon GPU Acceleration for d2l #2587

Open CharryLee0426 opened 4 months ago

CharryLee0426 commented 4 months ago

Description of changes: In this Pull Request: The Apple Silicon GPU support is added to d2l. So after this commit, d2l can detect Apple Silicon directly and use Apple Silicon GPU to accelerate parallel computing. It applies to two most popular deep learning frameworks: PyTorch and TensorFlow.

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

CharryLee0426 commented 4 months ago

If anyone interested in this pull request, look at issue#2588