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