【OpenMMLab 实战营打卡-第3课】计算机视觉之图像分类代码教学
使用 pip 安装的torch 不包括 cuda，所以需要使⽤ module 加载 cuda/11.1 模块。由于超算上原本编译器版本低于5，因此需要module命令加载一个高版本的gcc，本教程指定gcc版本为7.3。由于超算上使用mim命令安装可能存在兼容性的问题，因此这里使用pip命令安装mmcv-full模块。将数据集按照 8:2 的比例划分成训练和验证子数据集，并将数据集整理成 ImageNet的格式。代码如下：注意红框标出路径是否和自己本地路径一致，以自己的文件路径为准。
