centertrack torch1.11编译DCN

安装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已经编译成功了

centertrack torch1.11编译DCN_第1张图片

运行项目报错

centertrack torch1.11编译DCN_第2张图片

修改代码

在这里插入图片描述

  • 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++

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