Basic autocast usage

10

torch.cuda.amp is available in the nightly binaries and the current master.
If you are using an older version, you might need to update.

1 Like