can this model run on CUDA11.1?

prannaykaul / lvc

Official repo for our CVPR 22 paper: "Label, Verify, Correct: A Simple Few-Shot Object Detection Method"

Apache License 2.0

88 stars 14 forks source link

Open HefengRAY opened 2 years ago

HefengRAY commented 2 years ago

my system is working on a specific environment Python 3.7 torch 1.10.0 torchvision 0.11.1 cudatoolkit 11.1

but i find detetron2 version is 0.21 only work on cuda<=10.2

rustalter commented 1 year ago

CUDA11.1 is ok but I think python3.8 is necessary