在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) |