Phát hành PyTorch 1.8
(pytorch.org)PyTorch 1.8 đã được phát hành vào ngày 4 tháng 3 năm 2021.
-
Bổ sung API mới, cập nhật các tính năng học phân tán
-
Hỗ trợ cả ROCm cho GPU AMD (Beta)
-
Cũng cung cấp các bản cập nhật cho TorchVision, TorchText, TorchAudio
-
Cung cấp hướng dẫn PyTorch Mobile (chạy PyTorch trên Android / iOS)
2 bình luận
Tôi vẫn thường xuyên theo dõi tin tức ở đây. Tôi cũng xin để lại một góp ý.
Xin lưu ý là trong PyTorch 1.8 đã được bổ sung tính năng memory fraction, cho phép chia sẻ GPU memory.
https://pytorch.org/docs/stable/cuda.html
torch.cuda.set_per_process_memory_fraction(fraction, device=None)
Cảm ơn bạn~! Mình không rành mảng machine learning lắm haha
Bạn tổng hợp như vậy thật hay. Mong sau này thỉnh thoảng lại được nhờ bạn nhé! ^^