【DINO】环境配置

1. DINO简介

作为一款基于Transformer性能强劲的计算机视觉算法，一经发布即受追捧，本文记录下在DINO官方代码在集群上的环境配置及训练自己的数据集过程。

DINO原文：https://arxiv.org/abs/2203.03605

DINO源代码：https://github.com/IDEA-Research/DINO

2. DINO环境配置

该部分推荐大家参看DINO源代码中点击进入的配置过程，一般只要一步步来，不会出错的

​

根据上图大致流程为：

1 创建虚拟环境

2 安装合适的 torch相关包，比如上图中划线部分，注意cuda版本一定要和您的GPU CUDA版本契合，或者您通过nvidia-smi 查看的 GPU driver 假设是11.2 原则上，您在装cudatoolkit