PyToch和CUDA的版本不能兼容
install pytorch and torchvision
检查对应版本的
https://pytorch.org/get-started/previous-versions/
比如CUDA10.1对应的如下
CUDA 10.1
pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html