尽管已安装了Anaconda解释器和Torch,但在VSCode中无法导入Torch。

4

我安装了Python38、Python37和Anaconda,操作系统是win10 x64。

我无法在Python38和Python37中安装torch, 但在Anaconda中安装成功。

系统环境变量“Path”包含以下内容:

  • C:\python38\Scripts\
  • C:\python38\
  • C:\python37\Scripts\
  • C:\python37\
  • C:\anaconda3\
  • C:\anaconda3\Scripts

在VSCode中,即使我选择了Python 3.8.3 base-conda解释器:

enter image description here

我仍然无法导入torch。但是,在使用Jupyter Notebook时,我可以导入torch:

import torch
print(torch.__path__)

['C:\\anaconda3\\lib\\site-packages\\torch']

如何在 VS Code 中导入 Torch 并选择正确的解释器?

3个回答

3
当我们导入模块时,需要在当前选定的环境中安装所需的模块。(Python 环境)。
我使用以下语句在基本的 conda 环境中安装了它:

conda install pytorch torchvision -c pytorch

enter image description here

我们需要关注 Jupyter Notebook 中所选择的内核:

enter image description here


0
使用下面的pip命令安装pytorch:
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

执行此操作,然后重新加载您的VSCode。如果错误仍然存在,请检查您是否在所选的解释器环境中安装了该模块。这对我来说似乎起作用了。


0
我的解决方法是:卸载Python 3.7和3.8版本,只使用Anaconda作为所有Python库的来源。在导入torch后,你可能会遇到numpy库的错误。
from numpy.core._multiarray_umath import (
ImportError: DLL load failed: The specified module could not be found.

要修复它,需要添加另一个系统变量:

C:\anaconda3\Library\bin

其实也许我并不需要删除其他的Python版本。 但是毕竟,在VS Code中导入torch成功了:

from platform import python_version
print(python_version())
import torch
import torchvision
import numpy as np
print(torch.__path__)
print(np.__version__)

输出:

3.8.3
['C:\\anaconda3\\lib\\site-packages\\torch']
1.18.5

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接