import argparseimport timefrom pathlib import Pathimport cv2import torchimport torchbackendscudnn as cudnnfrom numpy import randomfrom modelsexperimental import attempt_loadfrom utilsdatasets import L

这个问题可能是因为PyTorch版本不兼容导致的。建议升级PyTorch版本到最新版，或者将PyTorch版本降级到与yolov5兼容的版本。可以使用以下命令升级PyTorch：

pip install torch --upgrade

或者降级PyTorch：

pip install torch==1.7.1

如果还是出现问题，可以尝试删除build文件夹并重新编译yolov5：

rm -rf build
python -m torch.utils.ffi.create_extension build/libmy_lib \
--include-path /usr/local/cuda/include --library-path /usr/local/cuda/lib64/ \
--extra-cflags '-std=c99' --extra-cflags -D_FORCE_INLINES \
--sources src/my_lib.c src/my_lib_cuda.c