- Melalui kolaborasi dengan tim pengembang Metal Apple, PyTorch kini dapat memanfaatkan akselerasi GPU di Mac
- Menggunakan MPS (Metal Performance Shader) sebagai backend PyTorch
- Dibandingkan saat menggunakan CPU dengan VGG16, performa meningkat 8x untuk training dan 21x untuk evaluation
2 komentar
Saya tambahkan juga secara singkat tulisan dari komunitas pengguna PyTorch Korea ^^;
https://discuss.pytorch.kr/t/apple-m1-pytorch-gpu/276
Saya merangkum cara menginstal PyTorch 1.12 di perangkat Apple M1 dan menggunakan perangkat MPS.
Karena
condatidak mendukung arm64, Anda mungkin mengalami kesulitan saat menggunakannya, jadi saya juga merangkum cara memeriksa dukungannya serta metode untuk mengatasinya.https://discuss.pytorch.kr/t/apple-m1-gpu/286