DINO-ubuntu20.04-单机多卡，训练自己的数据集

DINO-ubuntu20.04-单机6卡/3卡训练自己的数据集

---

---

前言

环境：ubuntu=20.04, python=3.7.3, torch=1.9.0, cuda=11.1

---

一、代码及数据准备

1、将自己数据转为-coco格式数据

COCODIR/
  ├── train2017/
  ├── val2017/
  └── annotations/
  	├── instances_train2017.json
  	└── instances_val2017.json

2、下载代码

git clone https://github.com/IDEA-Research/DINO.git
cd DINO

二、环境配置

1、创建新环境

conda create -n dino python=3.7 -y    # 新建环境
conda activate dino    # 激活环境

2、安装torch

在pytorch.org上找到合适自己cuda版本的pytorch安装即可，建议同源代码一样安装torch=1.9.0

pip install torch==1.9.0+cu111 torchvision==