3P by tkwlsrl 2021-03-14 | favorite | 댓글 2개

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)

고맙습니다~! 제가 머신러닝쪽은 약해서 ㅎㅎ
이렇게 정리해주시니 좋네요. 앞으로도 종종 부탁드려요! ^^