Colab训练YOLOv8遇RuntimeError及PyCharm运行报错的解决咨询

一、解决Colab训练时的RuntimeWarning #

/usr/lib/python3.10/multiprocessing/popen_fork.py:66: RuntimeWarning: os.fork() was called. os.fork() is incompatible with multithreaded code, and JAX is multithreaded, so this will likely lead to a deadlock

这个警告源于YOLOv8默认用fork启动多进程训练，而JAX的多线程机制与fork存在兼容性冲突，可通过以下方式解决：

• 修改训练代码的workers参数为0，禁用多进程数据加载：

  model.train(data='your_data.yaml', epochs=50, workers=0)
  

• 强制multiprocessing使用spawn启动方式，在训练代码开头添加：

  import multiprocessing
  multiprocessing.set_start_method('spawn', force=True)
  

二、解决PyCharm运行时的ultralytics.nn.conv报错 #

这类报错大概率是Colab训练环境与本地PyCharm环境的ultralytics版本不匹配，或训练生成的权重文件损坏导致，可按以下步骤修复：

• 同步版本：在Colab执行pip show ultralytics查看版本号，在PyCharm环境安装相同版本：

  pip install ultralytics==x.x.x  # 替换为Colab中的具体版本号
  

• 重新导出权重：若版本同步后仍报错，回到Colab重新完成训练并导出权重，避免训练中断导致文件损坏。
• 对齐底层依赖：确认PyCharm环境中torch、torchvision的版本与Colab一致，消除框架版本差异引发的算子兼容问题。

内容的提问来源于stack exchange，提问作者Faez AlABDALLAH ALHASNO