如何在Google Colab中更换PyTorch版本,安装后版本不匹配问题咨询
版本不匹配的核心原因
- 路径优先级问题
你使用sys.path.append()将conda的包路径添加到了sys.path的末尾,而Google Colab环境预装的PyTorch对应的路径优先级更高,排在路径列表的前面。Python导入包时会按顺序扫描sys.path里的路径,先找到哪个版本就加载哪个,所以你实际导入的是Colab预装的1.9+cuda120版本,而不是conda安装的1.0.0版本。 - 包管理工具的扫描范围差异
你用conda安装的PyTorch1.0.0确实已经成功写入到/usr/local/lib/python3.6/site-packages路径下,pip扫描时会遍历所有已登记的包路径,所以能识别到这个1.0.0的安装包,卸载时就会提示卸载该版本,和import时加载的预装版本不冲突。 - Python版本兼容问题
你安装的Miniconda3-4.5.4对应的默认Python版本为3.6,而当前Google Colab的默认Python版本已经远高于3.6,版本不一致也会导致conda安装的包无法被优先识别,甚至可能出现导入报错的问题。
解决方法
- 调整路径优先级:将append改为insert把conda路径放到最开头,代码如下:
import sys sys.path.insert(0, "/usr/local/lib/python3.6/site-packages")
- 更简单的方案是放弃安装Anaconda,直接用pip指定版本安装对应PyTorch即可,避免路径冲突。
- 安装完成后可以执行
print(torch.__file__)查看导入的PyTorch包的实际文件路径,确认版本对应关系。
内容的提问来源于stack exchange,提问作者shoehang
相关产品推荐
相关产品推荐

