繁体   English   中英

尽管安装了 anaconda 解释器和火炬,但无法在 vscode 中导入火炬

[英]can't import torch in vscode despite having anaconda interpreter and torch installed

我安装了 python38,37 和 anaconda,操作系统 - win10,x64。

我无法在 py38,37 中安装 torch - 但将其安装在 anaconda 中。

系统环境变量“路径”具有:

  • C:\python38\脚本\
  • C:\python38\
  • C:\python37\脚本\
  • C:\python37\
  • C:\anaconda3\
  • C:\anaconda3\脚本

在 vscode 中,尽管我选择了解释器作为 Python 3.8.3 base-conda

在此处输入图像描述

我仍然无法导入手电筒。 但是当使用 jupyter notebook 时,我可以导入torch:

import torch
print(torch.__path__)

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

如何在 vs 代码中导入 torch 并选择正确的解释器?

当我们导入模块时,我们需要在当前选择的环境中安装需要的模块。( Python环境。)

我使用以下语句将它安装在基本 conda 环境中:

conda install pytorch torchvision -c pytorch

在此处输入图像描述

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

在此处输入图像描述

什么对我有用:卸载 py37,38。 仅将 anaconda 作为所有 python 库的来源。 导入 torch 之后 - 您可能会从 numpy lib 中得到错误:

from numpy.core._multiarray_umath import (
ImportError: DLL load failed: The specified module could not be found.

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

C:\anaconda3\Library\bin

实际上,也许我不需要删除其他 python 版本。 但毕竟,导入torch在vs代码中工作:

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

使用如下所示的 pip 命令安装 pytorch:

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

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

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM