search

amirbar / DETReg

Official implementation of the CVPR 2022 paper "DETReg: Unsupervised Pretraining with Region Priors for Object Detection".
https://amirbar.net/detreg
Apache License 2.0
336 stars 46 forks source link

getting CUDA error #67

Closed ShreshthaSingh20 closed 1 year ago

ShreshthaSingh20 commented 1 year ago

on running sh ./make.sh I am getting this error. The CUDA_HOME is None because of which it shows "cuda not available" image (8)

abhinav3 commented 1 year ago

Hi Guys, any one open to solve this?

amirbar commented 1 year ago

Related to your machine config, you need to set the right CUDA_HOME env param.