This repository is based on Ultralytics/yolov5, with adjustments to enable rotate prediction boxes.
113
stars
21
forks
source link
raise EnvironmentError('CUDA_HOME environment variable is not set. ' OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root. #25
when I run setup.py, I encounter this problem