×

签到

分享到微信

打开微信,使用扫一扫进入页面后,点击右上角菜单,

点击“发送给朋友”或“分享到朋友圈”完成分享

【PyTorch】yolov5 val.py代码报错 blade2023-10-16 18:43:09 回复 10 查看 使用求助
【PyTorch】yolov5 val.py代码报错
分享到:

【寒武纪硬件产品型号】必填*:
必填项,例如:MLU370


【使用操作系统】必填*:
例如:ubuntu


【使用驱动版本】必填*:
例如v5.10.17


【出错信息】必填*:

RuntimeError: Pinned memory requires CUDA. PyTorch splits its backend into two shared libraries: a CPU library and a CUDA library; this error has occurred because you are trying to use some CUDA functionality, but the CUDA library has not been loaded by the dynamic er for some reason.  The CUDA library MUST be loaded, EVEN IF you don't directly use any symbols from the CUDA library! One common culprit is a lack of -Wl,--no-as-needed in your arguments; many dynamic ers will delete dynamic library dependencies if you don't depend on any of their symbols.  You can check if this has occurred by using ldd on your binary to see if there is a dependency on *_cuda.so library.

【当前已做了哪些信息确认】选填:
例如交叉验证/CPU能运行/调试通过等


【相关日志文档】选填

output.txt


【出错代码链接】选填:
github的或gitee的代码的链接,

projects/pytorch_yolov5_train/yolov5/val.py · Cambricon/practices - 码云 - 开源中国 (gitee.com) 

版权所有 © 2024 寒武纪 Cambricon.com 备案/许可证号:京ICP备17003415号-1
关闭