通过 conda 安装 的 detectron2

2023-12-26 18:37:09

从 detectron2官网 发现预编译的版本最高支持 pytorch1.10、cuda11.3。（2023-12-26）

1、安装 conda 环境。

 conda create --name detectron2 python=3.8

2、安装 pytorch1.10 和 cuda11.3。

pip3 install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio==0.10.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

3、安装 detectron2。

python -m pip install detectron2 -f \
  https://dl.fbaipublicfiles.com/detectron2/wheels/cu113/torch1.10/index.html

4、测试。

(detectron2) ai200@ubuntu16:/dev_sdc/wujp_py$ python
Python 3.8.18 (default, Sep 11 2023, 13:40:15) 
[GCC 11.2.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import detectron2
>>>

文章来源:https://blog.csdn.net/blueblood7/article/details/135225531
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。