文章目录
前言
我是在已有cuda10.1环境的linux服务器中配置centernet环境的,所以没有配置cuda环境的小伙伴可以先参考其他博客配置好cuda环境,再来看这篇博客。-------郑重提醒:这篇博客绝对可以解决小伙伴们在配置centernet环境时所遇到的大部分问题,说多了都是泪啊!!!
1、Anaconda虚拟环境创建
新上手的小伙伴一定要创建虚拟环境,在虚拟环境中配置包,不然会出现各种各样的错误。这里我使用的是Anaconda3。
conda create --name CenterNet python=3.6//创建python环境为3.6的虚拟环境
conda activate CenterNet //进入该环境
2、安装pytorch1.7
在安装pytorch之前,一定要先查看自己cuda版本,然后下载对应版本的pytorch,不然在编译DCNv2的时候会报错:没有cuda支持。
废话不多说,开整!!!
2.1可以去官网https://pytorch.org/ 选择对应自己cuda版本的pytorch
但是这种方法下载困难,我尝试很久都连接不上官网给的地址,再说官网给你配置的pytorch版本一般是最新的,不满足我们的需求。
2.2可以下载离线文件,然后在本地安装
这个方法速度很快,但是我在后面会遇到问题,测试torch时报错,缺失文件。可能是因为我的服务器中安装多个虚拟环境,本地安装时,系统检测到一些文件已经安装后,就不给我新建的虚拟环境安装了,系统真是太懒了!!!
2.3使用清华镜像
这是我最终选择的方法,下载速度很快,可以达到Mb/s级别。
首先确定你是在新建的虚拟环境中,即已经运行了:
conda activate CenterNet //进入该环境
然后配置清华镜像:参考https://zhangkaifang.blog.csdn.net/article/details/86