安装torch==1.11.0
- conda install pytorch1.11.0 torchvision0.12.0 torchaudio==0.11.0
cudatoolkit=11.3 -c pytorch
参考连接:https://pytorch.org/get-started/previous-versions/
DCNv2 安装教程
下载编译文件:https://github.com/jinfagang/DCNv2_latest
- cd CenterTrack-master/src/lib/model/networks
- unzip DCNv2-master.zip
- mkdir CenterTrack-master/src/lib/model/networks/DCNv2
- cd DCNv2-master
- mv ./* CenterTrack-master/src/lib/model/networks/DCNv2
- sh ./make.sh
有如下结果说明你的DCN已经编译成功了
运行项目报错
修改代码
from torch.hub import load_state_dict_from_url
3060 cuda报错 RuntimeError: No CUDA GPUs are available解决方案
- torch.cuda.is_available()
gcc g++ 安装和卸载
- sudo apt-get remove gcc gcc-xx #可能有多个版本,都要删掉
- sudo apt-get remove g++