detectron2 demo cuda10.0 py3.6

本文档详细记录了在CUDA 10.0和Python 3.6环境下安装Detectron2的过程,包括安装步骤和可能遇到的问题。参照提供的参考资料,可以顺利完成Detectron2的搭建。

Okay. No machine is immortal. Here’s my 180394930208th installation … ==||

Installation

conda create -n detectron python=3.6 pip
conda activate detectron
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
pip install cython pyyaml==5.1
pip install -U 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'
pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu100/index.html
# `cudaxxx` depends on your cuda version. 
conda install -c menpo opencv

vi mydemo.py

# usage: python mydemo.py
import detectron2
from detectron2.utils.logger import setup_logger
setup_logger
### 解释 `from vision.fair.detectron2.demo.predictor import vision` 的含义 `from vision.fair.detectron2.demo.predictor import vision` 是 Python 中的导入语句,用于从指定的模块路径中导入特定的对象或子模块。具体来说,它试图从 `vision.fair.detectron2.demo.predictor` 模块中导入名为 `vision` 的内容。`vision` 可以是一个类、函数、变量或者另一个子模块。 不过,根据之前提到的报错 “No module named ‘vision’” 来看,这里可能存在问题,也许是 `vision` 模块没有正确安装,或者导入路径有误。例如原本 `detectron2/demo/demo.py` 代码 `from vision.fair.demo.py.demo.predictor import VisualizationDemo` 就可能存在路径引用错误,实际上 `VisualizationDemo` 就在 `demo` 文件夹下的 `predictor.py` 文件中,可能不需要前面多余的路径 [^1][^2]。 ### 解决 `ModuleNotFoundError: No module named 'detectron2.demo'` 错误 #### 1. 检查 Detectron2 安装情况 要保证 Detectron2 已经正确安装,并且版本兼容。可以尝试重新安装 Detectron2: ```bash pip uninstall detectron2 pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cuXX/torchYY/index.html ``` 这里的 `cuXX` 代表 CUDA 版本,`torchYY` 代表 PyTorch 版本。 #### 2. 检查 Python 路径 确保 Detectron2 的根目录已经添加到 Python 的搜索路径中。可以在 Python 脚本中添加以下代码: ```python import sys sys.path.append('/path/to/detectron2') ``` 把 `/path/to/detectron2` 替换成 Detectron2 实际的安装路径。 #### 3. 检查文件结构 确认 `detectron2/demo` 目录确实存在,并且包含 `__init__.py` 文件(Python 会把包含该文件的目录视为一个包)。如果没有 `__init__.py` 文件,可以创建一个空的文件: ```bash touch /path/to/detectron2/demo/__init__.py ``` #### 4. 检查代码中是否有拼写错误 仔细检查代码里所有涉及 `detectron2.demo` 的导入语句,确保没有拼写错误。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值