Vscode + PyTorch + Anaconda联动

发布于:2022-12-28 ⋅ 阅读:(722) ⋅ 点赞:(0)

刚利用Anaconda安装了PyTorch,因为平常写代码喜欢使用Vscode,所以想办法如何在Vscode中能够import torch

1.首先得在conda中安装好PyTorch,以及创建好虚拟环境,具体步骤可见我的另一篇文章

再检查pytorch是否下载好了:

顺便看一下我们的pytorch所在的路径:

1ab4e1fd017fe931cf5da61a0b8aa825.png

(这是在我电脑中的路径)

 再检查我们conda中的python解释器的路径:(其实就是你所创建的虚拟环境)

也就是说我的python解释器路径是 D:\AAna\envs\PyTorch\python.exe

然后也请确保你的anaconda相关文件已经被加入了系统环境变量(虽然我也不知道为什么),具体自行操作

2.打开Vscode,如果你原来就已经下载好了python3.x,且已经写过代码,那么我猜你直接import PyTorch的话会出现下图的报错

我们可以分析图片中的报错:

第一行:ModuleNotFoundError说明模块不存在,故产生错误

第三行:可以发现此时我们的python路径是D:\code\vscode\python\1.py

很明显,与我们的conda虚拟环境路径不符合(也就是,此时vscode用的是我D盘中的python解释器,而要用pytorch,必须是conda中虚拟环境的解释器)

3.知道了异常原因,我们就要来修改vscode中python解释器的路径:shift+ctrl+p

选择上方的: Python:Select Interpreter

 

 选择你的虚拟环境即可

再测试,成功则大功告成,失败则看下一个步骤

 

4.若没有成功,而是出现了下图异常,则要进行本步骤。(此步骤并不针对每个人,这是本人踩过的坑)

我原以为经过步骤三已经成功了,没想到又报错。

我先是尝试了其他up主推荐的更新numpy版本,但没用:

 还是不行

后来考虑到,有可能是 "base环境的numpy"与"pytorch环境的numpy"重突,也有可能是:"pytorch环境中不同版本的numpy冲突" 

因此产生了解决方法:

在base环境中依次输入:有y按y确定

pip uninstall numpy 
conda uninstall numpy

再再pytorch环境(你自己创建的虚拟环境)中依次输入:

pip uninstall numpy 
conda uninstall numpy

最后,在pytorch环境中重新下载:

pip install numpy 

就成功了!!!!

 


网站公告

今日签到

点亮在社区的每一天
去签到