Hi,
Today in kaggle I installed the Nightly build through the command as mentioned on the pytorch.org page: # conda install -y pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch-nightly -c conda-forge
However again i am getting the error:
No module named ‘torch.cuda.amp.autocast’
As a second thought: It might also be an installation issue in Kaggle