在使用YOLOV8 训练模型的过程中,报段错误(核心已转储)
是pytorch版本与cuda版本不匹配
使用nvcc -V 查看cuda版本,在查找对于的pytorch,torchvison,torchaudio
本机为ubuntu 20版本 英伟达 nvcc-V 后的cuda版本为11.7
pip install torch==1.13.0 torchvision==0.14.0 torchaudio==0.13.0
在使用YOLOV8 训练模型的过程中,报段错误(核心已转储)
是pytorch版本与cuda版本不匹配
使用nvcc -V 查看cuda版本,在查找对于的pytorch,torchvison,torchaudio
本机为ubuntu 20版本 英伟达 nvcc-V 后的cuda版本为11.7
pip install torch==1.13.0 torchvision==0.14.0 torchaudio==0.13.0