Torch 已安装，但我无法将其导入 Jupyter Notebook 的计算机视觉 Python 项目中

问题描述

我正在使用检测器 2 进行计算机视觉项目。我在安装 Torch 或将其导入我的 jupyter 笔记本时遇到问题。我正在使用运行 MACOS Catalina，python3 版本为 3.8.2 的 mac，我在我的开发环境中使用 Anaconda。在屏幕截图中，它说火炬已安装，但在导入时抛出错误

解决方法

您已在 torch

安装了 /usr/local/lib/python3.9/site-packages for python

但正如你自己所说，你是

在我的开发环境中使用 Anaconda

因此，如果您的 jupyter notebook 确实配置为使用与 anaconda 一起提供的 python 安装，那么您需要这样做

conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch

注意：

1. 通过运行确定您的 jupyter 实际使用的是哪个 python 解释器

   import sys; print(sys.executable)

2. 确保为正确的 torch 环境安装了 conda（上面的命令还会指明是哪一个）

3. 根据您的需要设置 cudatoolkit 版本。 official website上还有一个简单的界面，可以获取正确的命令