728x90 반응형 xla1 [DL] PyTorch: TPU사용하기 - XLA https://github.com/pytorch/xla GitHub - pytorch/xla: Enabling PyTorch on XLA Devices (e.g. Google TPU)Enabling PyTorch on XLA Devices (e.g. Google TPU). Contribute to pytorch/xla development by creating an account on GitHub.github.com 다음 문서도 참고할 것.https://github.com/pytorch/xla/blob/master/API_GUIDE.md TPU로 PyTorch를 사용하기 위해서는 PyTorch / XLA 를 이용한다. PyTorch/xla의 정의 다음과 같음:“intermediate backend ena.. 2024. 3. 21. 이전 1 다음 728x90 반응형