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 457 forks source link

Closed rohinarora closed 3 years ago

rohinarora commented 3 years ago

Does env.yml need update? I get the below error : No module named 'torch.cuda.amp'

rohinarora commented 3 years ago

Fixed by using pytorch 1.7