Removing dependency on NVIDIA apex
RylanSchaeffer opened this issue · 1 comments
RylanSchaeffer commented
PyTorch has moved away from NVIDIA apex to torch.cuda.amp. Can someone explain the changes necessary to remove apex as a dependency?
RylanSchaeffer commented
I just discovered this comment: #56 (comment)
It specifies how to remove apex but not how to replace it with torch.cuda.amp