Torch 已安装，但我无法将其导入 Jupyter 笔记本中的计算机视觉 python 项目

Question

我正在使用 detectron2 进行计算机视觉项目。我在安装 torch 或将其导入我的 jupyter notebook 时遇到问题。我正在开发一个 mac，即 运行 MACOS Catalina，3.8.2 中的 Python3 版本，我正在使用 Anaconda 作为我的开发环境。在屏幕截图中，它说 torch 已安装，但在导入时抛出错误

Answer 1

您已在 /usr/local/lib/python3.9/site-packages

为 python 安装了 torch

但正如你自己所说，你是

using Anaconda for my Development environment

因此，如果您的 jupyter notebook 确实配置为使用与 anaconda 一起安装的 python，那么您需要执行

conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch

注：

1. 通过 运行

   确定你的 jupyter 实际使用的 python 解释器

   import sys; print(sys.executable)

2. 确保为正确的 conda 环境安装 torch（以上命令还将指出是哪个环境）

3. 根据需要设置cudatoolkit版本。 official website上也有一个简单的接口来获取正确的命令