[深度学习][原创]yolov5在GTX1660Ti上训练nan解决方法

训练时候box ,obj,cls一直是nan安装任何版本pytorch无效,更换yolov5-6.0最新版本无效

Epoch   gpu_mem       box       obj       cls    labels  img_size
       0/273     5.64G       nan       nan       nan       113       640

解决方法:

安装torch==1.9.1+cu102和torchvision==0.10.1+cu102即可解决问题

你可能感兴趣的:(深度学习,深度学习,pytorch,人工智能)