export TORCH_CUDA_ARCH_LIST=“7.5”
降低算力
或者
MMCV_WITH_OPS=1 MMCV_CUDA_ARGS=‘gencode=arch=compute_80,code=sm_80’ pip install -e .
不知道cudatoolkit什麼時候支持安培架構的GPU?
export TORCH_CUDA_ARCH_LIST=“7.5”
降低算力
或者
MMCV_WITH_OPS=1 MMCV_CUDA_ARGS=‘gencode=arch=compute_80,code=sm_80’ pip install -e .
不知道cudatoolkit什麼時候支持安培架構的GPU?