Closed wangg12 closed 5 years ago
I guess you are actually using CUDA 9.2 and the requirement for minimum driver version is 396.xx. Sometimes conda will install cuda 9.2 with pytorch.
I checked torch.version.cuda
, it is 9.0.176
. And my other pytorch programs are working without any problem. It is strange and I wonder where mmdet or mmcv check the cuda driver and cuda version.
That's strange. mmdet and mmcv do not check the driver, only pytorch itself will check it.
I updated the driver version from 390.87 to 410.73 and the problem is solved.
@hellock just as you said, conda may install another version cuda? I have met import issue when use mmdetection test.py, so I doubt my cuda version is not compatible. I installed cuda9.2.148.1(with cudnn9.2_v7.3.1, driver410.66) which I can check by nvcc -V, but I saw CUDA version10 showed in Nvidia-smi interface, which I installed long time ago and I already uninstalled. my question is: actually I did not find any cuda10 in conda or other place, any way to confirm which version cuda I use?
When I tried to train mask-rcnn on coco, an unexpected error occurred. I was able to train other pytorch networks besides mmdection. My driver version: 390.87 CUDA version: V9.0.176