MMDetection+3090+cuda11.1+Pytorch1.7.0

MMDetection+3090成功安装的环境配置

  • GTX 3090
  • cuda11.1
  • Pytorch1.7.0
  • 安装mmcv:pip install mmcv-full==1.2.4 -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.7.0/index.html
  • MMDetection就按照这里安装,之后测试Faster RCNN没有问题

出现问题的配置:

  • Pytorch1.8.0
  • mmcv-fulll的cu111, torch1.8.0的对应版本
  • MMDetection测试Faster RCNN会报错:CUDA error: no kernel image is available for execution on the device. 但不应该是Pytorch1.8有问题,Pytorch1.8+cuda11.1在别的code上不会报这个错,所以看起来是mmcv的对应版本还不能完全兼容。

你可能感兴趣的:(Detection)