在GPU上用VisDrone数据集训练yolov5问题汇总

（1）新建虚拟环境yolov5test

安装cuda10.2对应版本torch并测试，成功导入。显卡型号为：NVIDIA GeForce 940MX。

pip install torch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0

（2） requirement.txt 注释掉torch>=1.7.0 torchvision>=0.8.1两行，默认导入是CPU版本的，会覆盖安装好的GPU版本的torch。

（3）编译器选择，选择创建好的虚拟环境。