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

Question

我正在使用detectron2进行计算机视觉项目。我在安装 torch 或将其导入我的 jupyter 笔记本时遇到问题。我正在使用 3.8.2 中运行 MACOS Catalina、Python3 版本的 Mac，并且我将 Anaconda 用于我的开发环境。在屏幕截图中，它说安装了torch，但是在导入它时会抛出错误

Answer 1

你已经torch为 python安装了/usr/local/lib/python3.9/site-packages

但正如你自己所说，你是

在我的开发环境中使用 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 版本。官网也有一个简单的界面，可以获取正确的命令