detectron2_conda_env.yml

name: detectron2
channels:
  - torch
  - default
  - conda-forge
dependencies:
  - python=3.7
  - cython
  - pycocotools
  - pytorch::pytorch
  - cudatoolkit=10.1
  - gcc_linux-64
  - gxx_linux-64
  - jupyter
  - seaborn
  - pandas
  - pytorch::torchvision
  - conda-forge::opencv>=4.1
  - pip
  - pip:
     # -  git+https://github.com/facebookresearch/detectron2.git not working ? missing cublas.h ?
     # workarround with pre-compiled detectron2: pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu101/index.html