PyTorch Lightning
Đây là framework giúp viết code train mô hình PyTorch gọn gàng hơn, tự lo phần lặp lại như chia dữ liệu, lưu checkpoint và chạy trên nhiều card đồ họa. Bạn train AI có thể viết một lần rồi chạy trên 1 GPU hay 10.000 GPU vẫn cùng một code, không phải sửa gì thêm. Bạn cần cài Python và PyTorch trước, sau đó cài thêm gói lightning.
Framework giúp viết code train mô hình PyTorch gọn gàng hơn, tự lo phần lặp lại như chia dữ liệu, lưu checkpoint và chạy trên nhiều card đồ họa. Bạn viết code train một lần rồi chạy trên 1 GPU hay hàng nghìn GPU vẫn cùng một code, không phải sửa gì thêm, kèm cả công cụ Fabric cho ai muốn tự kiểm soát chi tiết. Bạn cần cài Python và PyTorch trước, sau đó cài thêm gói lightning.
git clone https://github.com/Lightning-AI/pytorch-lightning.git