I try to install pytorch 1.6.0 with pip.

torch                     1.6.0+cu101               
torchvision               0.7.0+cu101               
cudatoolkit               10.1.243             h6bb024c_0    defaults

but I got a error:

scaler1 = torch.cuda.amp.GradScaler()

AttributeError: module ‘torch.cuda’ has no attribute ‘amp’