Thanks very much for that. It did not seem like I would have much luck updating my NVIDIA drivers as I already had the latest release for my card, so instead I have used the option to downgrade pytorch.
Seems the latest that I can get for Cuda 10.1 is pytorch 1.8.1, according to Previous PyTorch Versions | PyTorch
So I have now run
pip install torch==1.8.1+cu101 torchvision==0.9.1+cu101 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
Making progress now
>>> import torch
>>> torch.cuda.is_available()
False
At least it’s not giving me the error this time!