This repository has been archived by the owner on Oct 31, 2023. It is now read-only.

Removing dependency on NVIDIA apex #96

Open

RylanSchaeffer opened this issue Feb 4, 2022 · 1 comment

RylanSchaeffer commented Feb 4, 2022

PyTorch has moved away from NVIDIA apex to torch.cuda.amp. Can someone explain the changes necessary to remove apex as a dependency?

The text was updated successfully, but these errors were encountered:

Author

RylanSchaeffer commented Feb 4, 2022

I just discovered this comment: #56 (comment)

It specifies how to remove apex but not how to replace it with torch.cuda.amp

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.