复现CenterNet

不瘦8斤的妥球球饼

已于 2022-04-25 09:38:32 修改

阅读量1.8k

点赞数 5

文章标签： pytorch 深度学习 python

于 2022-02-25 20:10:49 首次发布

本文链接：https://blog.csdn.net/weixin_43570470/article/details/123139512

版权

本文档详细介绍了如何复现CenterNet模型，包括在anaconda环境下创建新环境、安装pytorch（针对CUDA 9.0）、安装COCOAPI、编译DCNv2及解决编译过程中遇到的问题，如权限错误、符号未定义等。

摘要由CSDN通过智能技术生成

一、INSTALL.md安装

参考：https://blog.csdn.net/weixin_38715903/article/details/98039181

0.用anaconda3创建一个干净的环境

conda create --name CenterNet python=3.6
'激活它'
conda activate CenterNet

1. 安装pytorch

我的cuda版本：9.0

A.按照官方的安装代码，我是无法安装成功的

1）‘pytorch-0.4.1-py36_py35_py27__9.0.176_7.1.2_2.tar.bz2’无法下载下来
2）没法一次性安装pytorch=0.4.1和torchvision

按照如下步骤：
1）卸载原来用anaconda3创建环境时自带的pytorch（可以先用1. import torch
2. print(torch.__version__)看看有没有已安装的高版本pytorch，再进行这一步），卸载步骤参考https://blog.csdn.net/mao_hui_fei/article/details/112078113，如果当初是用conda install 命令安装的pytorch，则用第一种方法，如果是用pip install命令安装的，则用第二种方法。或者两种方法都试一下。
1.1）使用conda卸载Pytorch
conda uninstall pytorch
conda uninstall libtorch
1.2）使用pip卸载Pytorch
pip uninstall torch
2）在https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/中下载
pytorch-0.4.1-py36_py35_py27__9.0.176_7.1.2_2.tar.bz2
3）conda install pytorch=0.4.1 -c pytorch
4）conda install =torchvision

B.And disable cudnn batch normalization

python3.6
import torch
print(torch.backends.cud

最低0.47元/天解锁文章