mmdetection配置踩坑

THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=405 error=11 : invalid argument 解決:由於時RTX2080Ti的卡,cuda需要10.0以上,在安裝pytoch時不能直接通過pip install torch來安裝,需要通過wheel下載對應cuda版本的pytorch才行,可在http
相關文章
相關標籤/搜索