![](https://img-blog.csdnimg.cn/20201014180756757.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
gpu
文章平均质量分 51
至尊宝♬
仍未成功,只因努力不够。
展开
-
【亲测通过】MaskRcnn_tf1.x如何升级到MaskRcnn_tf2.x,实现RTX3090环境训练自定义数据集模型。
一、背景:之前一篇博文中已经实现了maskrcnn_tf1.15.0环境的win10+cpu模型训练,但cpu训练实在是非常的耗时,据说tf1.x是支持RTX1060的(本人未测试),但不支持最新的RTX3090,查阅了很多资料,原因应该是tf1.x与tf2.x的差别比较大,必须升级到tf2.x,才可以正常使用rtx3090。下面是maskrcnn_tf1.15.0的开发案例,本人亲测可用。mask_rcnn 训练自定义数据集(本地win10系统cpu已调通,采坑无数,均已列出解决方法)_君..原创 2022-03-11 15:07:21 · 3474 阅读 · 35 评论 -
Internal: Invoking ptxas not supported on WindowsRelying on driver to perform ptx compilation.
环境:Windows 10, gpu 3090, TF 1.15,cuda_10.0.130_411.31_win10,cuDNN 7.6.5.32mask-rcnn框架,运行train.py报错信息如下:image_id 333image_id 32image_id 582022-01-01 19:03:07.415032: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully op..原创 2022-01-03 16:47:53 · 5022 阅读 · 4 评论 -
mask-rcnn报错: AttributeError: ‘Model‘ object has no attribute ‘metrics_tensors‘
环境:win10+gpu 3090执行解决方法:原创 2022-01-01 17:36:03 · 1263 阅读 · 2 评论 -
AttributeError: module ‘tensorflow._api.v1.compat.v2‘ has no attribute ‘__internal__‘【亲测有效】
环境:win10+gpu 3090mask-rcnn-master模型在执行train.py时报错:AttributeError: module 'tensorflow._api.v1.compat.v2' has no attribute '__internal__'解决方法:1.卸载:pip uninstall keras-nightly2.安装:pip install -i https://pypi.douban.com/simple/ h5py==2.10.0pi原创 2022-01-01 17:22:57 · 4144 阅读 · 5 评论 -
【已解决---->“重点在后面”】win10+gpu,keras-yolov3,执行train.py训练模型,loss: nan
keras-yolov3在win10+cpu下训练模型正常;但同样的代码在win10+gpu环境下,训练模型一直显示loss:nan。1.错误信息如下:Epoch 1/502021-12-24 18:31:54.894750: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudnn64_7.dll2021-12-24 18:31:57.8原创 2021-12-24 18:39:40 · 1324 阅读 · 1 评论 -
win10GPU服务器,keras-yolov3执行train.py时,提示Adding visible gpu devices: 0 卡着不动
解决方法:我的电脑,右键,属性,高级系统设置,高级,环境变量,然后添加一个CUDA_CACHE_MAXSIZE变量,值为:4294967296,点击确定原创 2021-12-23 13:56:44 · 1869 阅读 · 0 评论