首先要配置环境
conda create -n nanodet python=3.8 -y
conda activate nanodet
确认一下cuda版本
nvcc -V
确认是11.3之后,要安装11.3对应的pytorch版本。
本机装pytorch1.12.1后面运行的时候会报错(torch没有经过cuda编译),于是降级到1.11
conda install pytorch==1.11.0 torchvision==0.12.0
该文介绍了如何配置conda环境,安装CUDA和PyTorch,以及为NanoDet模型下载和准备COCO数据集。在完成环境配置后,文章详细阐述了如何训练NanoDet模型,包括修改配置文件,设置训练参数,以及在PyCharm中运行训练。最后,文章提到了如何测试训练好的模型并展示了50个epoch的训练结果。
首先要配置环境
conda create -n nanodet python=3.8 -y
conda activate nanodet
确认一下cuda版本
nvcc -V
确认是11.3之后,要安装11.3对应的pytorch版本。
本机装pytorch1.12.1后面运行的时候会报错(torch没有经过cuda编译),于是降级到1.11
conda install pytorch==1.11.0 torchvision==0.12.0
2129
1万+

被折叠的 条评论
为什么被折叠?