3 điểm bởi tkwlsrl 2021-03-14 | 2 bình luận | Chia sẻ qua WhatsApp

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

 
tkwlsrl 2021-03-14

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)

 
xguru 2021-03-14

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é! ^^