建议重新安装cu118版本的pytorch。
或者在https://download.pytorch.org/whl/nightly挑选你喜欢的torch和torchvision版本,注意先装torch后装torchvision