我之前的conda环境名是lzj,我在2022年9月21日安装的gpu版本的pytorch,其安装包位置如下图所示
由于最近手欠,手动输入了pip install torch,就默认安装了cpu版本的pytorch了,此时我再去跑以前的代码就会报错:报错的大概意思就是要用gpu版本的torch,但此时是cpu版本的torch
实际上,我认为,即使输入了pip install torch,conda也没有将cpu版的torch覆盖掉之前gpu版本的torch,原因在于之前安装gpu版本的torch时,我是用conda install安装的,所以conda安装的torch的位置如上图所示。而我们在conda环境下使用pip install安装时,其默认的安装位置是在下图所示,我的电脑是安装在C:\Users\Administrator\AppData\Roaming\Python\Python39\site-packages这个路径下,因为我的conda环境lzj用的是python3.9
我的解决方案非常简单,就是直接将该路径下的cpu版本的torch文件直接删除,不用担心conda无法import torch,因为我认为conda搜索某一个库的顺序是先去pip install的地方去找,就是上图所述C盘里的python39里面去找,如果这个里面没有,就会去anaconda文件夹自身的site-packages里面找了。