复现PointRCNN 中碰到RuntimeError: CUDA error: no kernel image is available for execution on the device

导致的原因一般都是显卡算力和cuda或者torch版本不匹配

由于我是自己去下载的torch和torchvision可能没下好对应的关系,于是我卸载了相对应的torch和torchvision,运行下面的命令  安装了对应的版本

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch

你可能感兴趣的:(python,目标检测)