在win10上build Detectron2报错:ValueError: Unknown CUDA arch (8.6) or GPU not supported

RTX30系列显卡与CUDA10版本不兼容

我的系统:win10
我的环境:CUDA11.1和CUDA10.1(这里用的是10.1版本)

配置detectron2时,运行到python setup.py build develop时,warning:

Warning: GeForce RTX 3070 with CUDA capability sm_86 is not compatible with the current PyTorch installation. The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37. If you want to use the GeForce RTX 3070 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/local

报错:

ValueError: Unknown CUDA arch (8.6) or GPU not supported

说白了就是,CUDA版本过低,GPU版本过高,不匹配
需换用高版本的CUDA

参考文章

你可能感兴趣的:(cuda,gpu,深度学习,pytorch)