Anaconda
在清华镜像中下载Anaconda,我们下载的版本是
Anaconda3-2021.11-Windows-x86_64.exe 之前将Anaconda安装到了D盘,要卸载干净后重新安装到了C盘下。安装时有一个添加进环境变量的配置要记得勾选。
cuda11.6
cuda官网默认的是11.7,为配合pytorch需要安装cuda11.6版本。
cuda安装时点“自定义”即可。我们的默认是cuda17,没卸载,需要在用cuda 的时候标记cu116。
cuda版本查看:
nvidia-smi
pytorch安装步骤
根据cuda版本安装pytorch,必须Python版本+PyTorch版本+torchversion版本+torchaudio版本+cuda版本完全一一对应才能正常使用。
python版本是3.9,cuda是11.6,pytorch官网选择
直接运行conmand报错,因为网络有些拉跨,就采用将后面的网址直接用浏览器打开,将原xommand中要安装的torch、 torchvision 和torchaudio依次下载
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116
https://download.pytorch.org/whl/cu116
会看到一溜.whl文件, 挑一个版本下载。根据我们的配置,搜索“cu116-cp39-cp39‘,下载对应的版本。cu116就是cuda11.6,cp39就是python3.9.x缩写。
torchvision:
torchandio:
torch:
下载好后,在whl文件的文件夹中打开命令行,运行
pip install 对应的whl文件的全称
测试:
python
>import torch
>torch.cuda.is_available()
返回True那么就是好了,可以跑显卡了。如果不行那么就是没安装对,只能用cpu跑。
yolov5代码
1.数据集存一个images、一个labels文件夹,labels转为txt格式
2.改一个coco.yaml,主要改:
a.train、val、test路径
b.改nc、names;nc表示数据集类别,那么时表示数据集标注的类名称
3.可以修改models模型文件,我们没改
4.训练train.py
weights、cfg、data按照文件路径修改。
从cfg默认初始models/yolovs.yaml,但我们训练的时候设为‘’
epochs地带次数自己决定,我们默认的是300
5.训练中止时,exp/weights下会有best和last两个pt。继续开始训练只需要将last.pt路径写到weights下面/
6.测试时,只需要更改下面参数的路径即可
nice!