Colab中安装PyTorch 2.0.0后导入版本不符问题求助

Colab默认预装了特定版本的PyTorch，即便你通过pip install --upgrade命令成功安装了2.0.0，当前运行时仍会使用之前加载到内存里的旧版本，必须重启运行时才能切换到新安装的版本。

解决步骤：

• 执行升级命令后，点击Colab界面顶部的「Runtime」→「Restart runtime」，重启完成后重新运行导入和版本查看代码。
• 如果重启后依旧不生效，直接指定对应版本号强制重装（注意PyTorch、torchvision、torchaudio版本需匹配）：

  !pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --force-reinstall
  

  安装完成后再次重启运行时，然后执行以下代码验证：

  import torch
  print(torch.__version__)
  

• 还可以检查PyTorch的安装路径，确认新版本是否正确安装：

  import torch
  print(torch.__file__)
  

  若输出路径指向site-packages下的PyTorch 2.0.0目录，说明安装成功；若不是，重启运行时通常就能解决路径优先级问题。

内容的提问来源于stack exchange，提问作者d3dalo