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
参考文章