BraveGroup/PSPS

pytorh,cuda

iskyiminever opened this issue · 1 comments

Hello, can you tell me which version torch and cuda need?

js-fan commented

Hi, this code is tested with Python 3.7 + Pytorch 1.10(cu111) and Python 3.10 + Pytorch 1.13(cu117). Please ensure the mmdet is imported from the repo instead of pre-installed packages.