SimCLR icon indicating copy to clipboard operation
SimCLR copied to clipboard
verified publisher icon sthalles

No module named 'torch.cuda.amp'

Open canqin001 opened this issue 4 years ago • 2 comments

When I run "python run.py", there is an error as No module named 'torch.cuda.amp'. Does anyone know how to address it?

canqin001 avatar Jul 22 '21 17:07 canqin001

Use pytorch >=1.6.0

Mrharry1997 avatar Jul 28 '21 07:07 Mrharry1997

update your pytorch

LifeIsBright-heihei avatar Oct 12 '21 06:10 LifeIsBright-heihei