PyTorch 1.8 Release
(pytorch.org)2021년 3월 4일 pytorch 1.8 이 릴리즈 되었습니다.
- 새로운 API 추가, 분산 학습 기능 업데이트
- AMD GPU를 위한 ROCm 도 지원 ( Beta )
- TorchVision, TorchText, TorchAudio 의 업데이트도 제공
- PyTorch Mobile 튜토리얼제공(Andorid / iOS 에서 PyTorch 실행 )
평소에 뉴스 잘 보고 있습니다. 저도 하나 남겨봅니다.
참고로 PyTorch 1.8에서 GPU memory를 나눠 쓸 수 있는 memory fraction 기능이 추가 되었습니다.
https://pytorch.org/docs/stable/cuda.html
torch.cuda.set_per_process_memory_fraction(fraction, device=None)