深度学习常用操作笔记

指令

nvidia-smi
pip install torch1.12.1+cu116 torchvision0.13.1+cu116 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu116
python tools/train.py configs/top_down/dite_hrnet/coco/ditehrnet_18_coco_256x192.py

报错

cannot import name ‘Config’ from ‘mmcv’

根据 mmcv 的官方文档，在 2.0.0 及更高版本中，许多模块已迁移到新的 mmengine 库中。因此，Config 等功能需要从 mmengine 而非 mmcv 中导入。
修改 tools/train.py 脚本中的相关导入，将 mmcv 的部分导入更换为 mmengine：

# 将以下代码
# from mmcv import Config, DictAction# 替换为
from mmengine.config import Config
from mmengine.utils import DictAction

ImportError: cannot import name ‘print_log’ from ‘mmcv’

找到以下导入语句：

from mmcv.runner import init_dist, set_random_seed
修改为：from mmengine.runner import init_dist
from mmengine.runner import set_random_seed
from mmengine.logging import print_log

ImportError: cannot import name ‘init_dist’ from ‘mmengine.runner’

在 mmengine 的最新版本中，init_dist 函数通常已迁移到 mmengine.distributed。
将导入语句：

#from mmengine.runner import init_dist
#替换为：
from mmengine.dist import init_dist

WARNING: Retrying (Retry(total=4, connect=None, read=None, redirect=None, status=None)) after connection broken by ‘SSLError(SSLZeroReturnError(6, ‘TLS/SSL connection has been closed (EOF) (_ssl.c:1149)’))’: /simple/mmdet/

网络问题，关闭梯子或者代理

总结

太痛苦了，两天总是报各种各样的错误，真特么气死人。无穷无尽的bug，本来还想解决跑下别人的深度学习模型，结果无穷无尽的bug，真该死。