No module named 'torch.cuda.amp'

sthalles / SimCLR

PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations

MIT License

2.19k stars 458 forks source link

Open canqin001 opened 3 years ago

canqin001 commented 3 years ago

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

Mrharry1997 commented 3 years ago

Use pytorch >=1.6.0

LifeIsBright-heihei commented 2 years ago

update your pytorch