AlexeyAB / darknet

YOLOv4 / Scaled-YOLOv4 / YOLO - Neural Networks for Object Detection (Windows and Linux version of Darknet )
http://pjreddie.com/darknet/
Other
21.6k stars 7.95k forks source link

cuda11.0 support (RTX3080) #6827

Open wyukai opened 3 years ago

wyukai commented 3 years ago

Hello, may I ask when CUDA11 (RTX3080) can be supported?

johnny-mueller commented 3 years ago

I am currently using CUDA 11 and therefore it is already supported. But as far as I can see, some have the problem with the latest RTX graphics cards. This seems to be independent of the CUDA version.

Wenlong0605 commented 3 years ago

Did you use RTX3080 with CUDA11.1 and cudnn8 to train yolov4 or yolov3? @johnny-mueller

LeKristapino commented 3 years ago

@Wenlong0605 Cuda 11.1 and cudnn8 works fine with RTX 3090 and yolov4. Shouldn't be a problem with 3080

liaobear commented 3 years ago

hey, i tried Cuda 11.0~11.1 and cudnn8 with RTX 3080 , darknet test can use GPU ,but when i try to train,it actually use cpu to train,no use gpu.Can anyone solve?