1.报错如下:
2.让我们进入yolov5项目,找到export.py文件,打开
1. 我的是在355行左右,可以ctrl+F检索
2. 把第3行代码注释或删除,也就是下面的代码
config.max_workspace_size = workspace * 1 << 30
3. 再添加上如下代码,就可以运行啦
config.max_workspace_size = 1 << 20
PS:如果解决了,记得给博主点赞哦
1. 我的是在355行左右,可以ctrl+F检索
2. 把第3行代码注释或删除,也就是下面的代码
config.max_workspace_size = workspace * 1 << 30
3. 再添加上如下代码,就可以运行啦
config.max_workspace_size = 1 << 20
PS:如果解决了,记得给博主点赞哦