更新torch的版本,原先是1.4.0出现这个问题,更新到1.11.0后解决。
pip install torch==1.11.0+cpu torchvision==0.12.0+cpu torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cpu