▲
tkwlsrl 2021-03-14 | parent | ★ favorite | on: PyTorch 1.8 Release(pytorch.org)

평소에 뉴스 잘 보고 있습니다. 저도 하나 남겨봅니다.
참고로 PyTorch 1.8에서 GPU memory를 나눠 쓸 수 있는 memory fraction 기능이 추가 되었습니다.
https://pytorch.org/docs/stable/cuda.html
torch.cuda.set_per_process_memory_fraction(fraction, device=None)

▲
xguru 2021-03-14  [-]

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

답변달기