Detectron2 安装踩坑

在服务器上安装Detectron2:
设备:NVIDIA GeForce RTX 3090
显卡驱动和cuda:在这里插入图片描述
安装detectron2时,报错为:

Installing collected packages: detectron2
  Running setup.py develop for detectron2
    error: subprocess-exited-with-error
    
    × python setup.py develop did not run successfully.
    │ exit code: 1
    ╰─> [59 lines of output]
        running develop
        running egg_info
        creating detectron2.egg-info
        writing detectron2.egg-info/PKG-INFO
        writing dependency_links to detectron2.egg-info/dependency_links.txt
        writing requirements to detectron2.egg-info/requires.txt
        writing top-level names to detectron2.egg-info/top_level.txt
		......
      self.initialize_options()
    error: command '/usr/local/cuda/bin/nvcc' failed with exit code 1
    [end of output]

note: This error originates from a subprocess, and is likely not a problem with pip.

在网上查找问题说是gcc与cuda版本匹配问题,因为没搞懂升级gcc,就升级cuda之后:
在这里插入图片描述
可以正常安装。
如有错误敬请指正!!!
查询了一些博客:
detectron2安装(亲测好用)-试了一下没有成功

你可能感兴趣的:(pytorch,人工智能)