PyTorch, Mac (Apple Silicon M1) पर GPU acceleration सपोर्ट

• Apple की Metal development team के साथ सहयोग से अब PyTorch को Mac पर GPU acceleration मिल गया है
  • PyTorch के backend के रूप में MPS (Metal Performance Shader) का उपयोग
• VGG16 के साथ CPU उपयोग की तुलना में training में 8 गुना और evaluation में 21 गुना performance improvement देखा गया

[ninebow]

हल्के से PyTorch Korea User Group की पोस्ट भी जोड़ रहा/रही हूँ ^^;

https://discuss.pytorch.kr/t/apple-m1-pytorch-gpu/276

[ninebow]

Apple M1 डिवाइस पर PyTorch 1.12 इंस्टॉल करने और MPS डिवाइस का उपयोग करने का तरीका संक्षेप में बताया गया है.

conda में arm64 सपोर्ट न होने के कारण उपयोग में कठिनाई हो सकती है, इसलिए सपोर्ट उपलब्धता की जांच और उसके समाधान के तरीके भी साथ में व्यवस्थित किए गए हैं.

https://discuss.pytorch.kr/t/apple-m1-gpu/286