cuda10.1+pytorch1.7配置centernet环境

最新推荐文章于 2024-04-05 17:37:37 发布

二水笔记

最新推荐文章于 2024-04-05 17:37:37 发布

阅读量3.4k

点赞数 3

分类专栏： CenterNet 文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/weixin_43786395/article/details/122195610

版权

前言

我是在已有cuda10.1环境的linux服务器中配置centernet环境的，所以没有配置cuda环境的小伙伴可以先参考其他博客配置好cuda环境，再来看这篇博客。-------郑重提醒：这篇博客绝对可以解决小伙伴们在配置centernet环境时所遇到的大部分问题，说多了都是泪啊！！！

新上手的小伙伴一定要创建虚拟环境，在虚拟环境中配置包，不然会出现各种各样的错误。这里我使用的是Anaconda3。

conda create --name CenterNet python=3.6//创建python环境为3.6的虚拟环境
conda activate CenterNet //进入该环境

在安装pytorch之前，一定要先查看自己cuda版本，然后下载对应版本的pytorch，不然在编译DCNv2的时候会报错：没有cuda支持。
废话不多说，开整！！！

但是这种方法下载困难，我尝试很久都连接不上官网给的地址，再说官网给你配置的pytorch版本一般是最新的，不满足我们的需求。

这个方法速度很快，但是我在后面会遇到问题，测试torch时报错，缺失文件。可能是因为我的服务器中安装多个虚拟环境，本地安装时，系统检测到一些文件已经安装后，就不给我新建的虚拟环境安装了，系统真是太懒了！！！

这是我最终选择的方法，下载速度很快，可以达到Mb/s级别。
首先确定你是在新建的虚拟环境中，即已经运行了：

conda activate CenterNet //进入该环境

然后配置清华镜像：参考https://zhangkaifang.blog.csdn.net/article/details/86497684

关注