PyTorch hỗ trợ tăng tốc GPU trên Mac (Apple Silicon M1)

• Nhờ hợp tác với đội ngũ phát triển Metal của Apple, PyTorch nay có thể tăng tốc GPU trên Mac
  • Sử dụng MPS (Metal Performance Shaders) làm backend cho PyTorch
• Khi so sánh với việc dùng CPU trên VGG16, hiệu năng được cải thiện 8 lần cho huấn luyện và 21 lần cho đánh giá

[ninebow]

Xin nhẹ nhàng bổ sung thêm bài viết từ cộng đồng người dùng PyTorch Hàn Quốc ^^;

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

[ninebow]

Đã tổng hợp cách cài đặt PyTorch 1.12 trên thiết bị Apple M1 và sử dụng thiết bị MPS.

Do conda không hỗ trợ arm64 nên việc sử dụng có thể gặp khó khăn; bài viết cũng tổng hợp cách kiểm tra việc hỗ trợ và các phương án khắc phục.

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