Chinaunix首页 | 论坛 | 博客
  • 博客访问: 723310
  • 博文数量: 118
  • 博客积分: 1437
  • 博客等级: 上尉
  • 技术积分: 1155
  • 用 户 组: 普通用户
  • 注册时间: 2009-02-22 20:23
文章分类

全部博文(118)

文章存档

2022年(32)

2017年(3)

2014年(4)

2013年(1)

2011年(2)

2010年(16)

2009年(60)

我的朋友

分类: LINUX

2022-01-26 19:56:59

在win10下调试CenterNet遇阻,总是出现显卡的错误。尝试在Ubuntu16.04下重新配置环境,调试CenterNet.

1. CUDA版本
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Sat_Aug_25_21:08:01_CDT_2018
Cuda compilation tools, release 10.0, V10.0.130

2.安装Anaconda及PyCharm

3.安装PyTorch
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch 安装失败

pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f 
-f 

Successfully installed torch-1.7.1+cu101 torchaudio-0.7.2 torchvision-0.8.2+cu101
安装成功

4. 参照配置centerNet
https://blog.csdn.net/babyzbb636/article/details/102544507

5. 下载pytorch 1.7对应的DCNv2



6.遇到错误No module named 'models.networks.DCNv2._ext'

原因:未将DCNv2放入正确的路径 src\lib\models\networks\DCNv2








阅读(1111) | 评论(0) | 转发(0) |
给主人留下些什么吧!~~