TorchStudio / torchstudio

IDE for PyTorch and its ecosystem
https://torchstudio.ai
MIT License
378 stars 27 forks source link

RuntimeError: Currently, AutocastCPU only support Bfloat16 as the autocast_cpu_dtype #41

Closed lbgitjp closed 1 year ago

lbgitjp commented 1 year ago

I followed the steps from here: https://www.torchstudio.ai/getstarted/,

But at the last step shows a error .

torch.set_autocast_cpu_dtype(self.fast_dtype) # type: ignore[arg-type] RuntimeError: Currently, AutocastCPU only support Bfloat16 as the autocast_cpu_dtype

please see below pic: 1

log file: TorchStudio 2022.10.10-20.09.33.txt

please tell me how to resolve it?

thanks!

divideconcept commented 1 year ago

This is now fixed with 0.9.12: https://github.com/TorchStudio/torchstudio/releases/tag/0.9.12

lbgitjp commented 1 year ago

It works, thank you!