PyTorch รองรับการเร่งความเร็วด้วย GPU บน Mac (Apple Silicon M1)

 (pytorch.org)
12 คะแนน โดย xguru 2022-05-20 | 2 ความคิดเห็น | แชร์ทาง WhatsApp
  • PyTorch สามารถเร่งความเร็วด้วย GPU บน Mac ได้แล้ว จากความร่วมมือกับทีมพัฒนา Metal ของ Apple
    • ใช้ MPS (Metal Performance Shader) เป็นแบ็กเอนด์ของ PyTorch
  • เมื่อเทียบกับการใช้ CPU โดยใช้ VGG16 พบว่า Training เร็วขึ้น 8 เท่า และ Evaluation เร็วขึ้น 21 เท่า

บทความที่เกี่ยวข้อง

2 ความคิดเห็น

 
ninebow 2022-05-20

ขอแนบโพสต์จากชุมชนผู้ใช้ PyTorch เกาหลีมาเบา ๆ ด้วยครับ ^^;

https://discuss.pytorch.kr/t/apple-m1-pytorch-gpu/276
 
ninebow 2022-05-21

ได้สรุปวิธีติดตั้ง PyTorch 1.12 บนอุปกรณ์ Apple M1 และการใช้งานอุปกรณ์ MPS สำหรับเร่งความเร็วด้วย GPU บน Mac (Apple Silicon M1)

เนื่องจาก conda ยังไม่รองรับ arm64 จึงอาจทำให้ใช้งานได้ยาก จึงได้รวบรวมวิธีตรวจสอบการรองรับและแนวทางแก้ไขไว้ด้วย

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