search

NVlabs / DeepInversion

Official PyTorch implementation of Dreaming to Distill: Data-free Knowledge Transfer via DeepInversion (CVPR 2020)
Other
485 stars 77 forks source link

AttributeError: module 'torch.cuda.amp' has no attribute 'initialize' #28

Open Egg-Hu opened 1 year ago

Egg-Hu commented 1 year ago

AttributeError: module 'torch.cuda.amp' has no attribute 'initialize'

xmutly commented 10 months ago

I met same error, maybe the vesion of your torch is too new.

Wings-Of-Disaster commented 2 months ago

You need to install APEX library