各位大佬好,我想跑YOLOV5,用极链云租了个实例,
按照帮助文档:https://cloud.videojj.com/help/。配置好了环境,pycharm deployment也配置成功了,可以与服务器进行文件上下载和比较。
于是我运行train.py,网络构建好了,已经读取数据了,进入第0个epoch。然后没过多久,就提示报错了!
报错:
THCudaCheck FAIL file=/opt/conda/conda-bld/python/pytorch/work/aten/src/THC/THCCachingHostAllocator.cpp line=278 error=304
报错:
RuntimeError: Caught RuntimeError in pin memory thread for device 0.
O
THCudaCheck FAIL file=/opt/conda/conda-bld/python/pytorch/work/aten/src/THC/THCCachingHostAllocator.
于 2022-02-16 15:55:13 首次发布