detectron2安装方法
安装detectron2
·
安装detectron2
安装detectron2的方法,官方教程:Installation — detectron2 0.6 documentation
第一种方法:Build Detectron2 from Source
官方大大的源码:
python -m pip install 'git+https://github.com/facebookresearch/detectron2.git'
# (add --user if you don't have permission)
# Or, to install it from a local clone:
git clone https://github.com/facebookresearch/detectron2.git
python -m pip install -e detectron2
# On macOS, you may need to prepend the above commands with a few environment variables:
CC=clang CXX=clang++ ARCHFLAGS="-arch x86_64" python -m pip install ...
注意:如果采用先下载detectron2,再安装的方法,下载之后不要 cd detectron2,否则会出现找不到“step.py”的错误。
我的环境是CUDA:11.6, pytorch:1.10.2
采用第一种方式安装,会出现CUDA与torch版本不匹配。报错如下:
第二种方法:Install Pre-Built Detectron2 (Linux only)
知道自己的CUDA版本和torch版本后,采用第二种方法更快,而且不易出错,我就是采用的这种方法安装完成的。
我的环境是CUDA:11.6, pytorch:1.10.2,我选择的是CUDA==10.2, torch==1.10。
成功!
更多推荐
已为社区贡献2条内容
所有评论(0)