Tech News

https://pytorch.org/blog/pytorch-1.8-released/ 

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)