RAFT icon indicating copy to clipboard operation
RAFT copied to clipboard

In cuda11.0, running time become slower than cuda10.x

Open JunHeum opened this issue 4 years ago • 3 comments

Hi,

I found that using CUDA11.0+PyTorch1.7.0+CUDNN8 almost doubles the running time(train and test) compared to using CUDA10.2+PyTorch1.7.0+CUDNN7.

To use RTX 30 series graphic cards, we must use CUDA11 and CUDNN 8.

When you afford to experiment, I am wondering if I could know what parts in your codes make this issue.

JunHeum avatar Dec 10 '20 13:12 JunHeum

Hi, I will look into this in the next few days

zachteed avatar Dec 10 '20 16:12 zachteed

I would like to know if there is a better solution

gaoyumin11 avatar Jun 10 '21 12:06 gaoyumin11

I found out that this is due to the num worker number being set too small.

MiaoJieF avatar Jun 27 '22 08:06 MiaoJieF