- 博客(3)
- 收藏
- 关注
RSS订阅原创 Tensorflow 2.1+Pycharm明明GPU没有被占用却遇到Out of Memory问题
GPU没有被占用时提示Out of Memory记录在开头添加这一段 记录 在开头添加这一段 // #------------------config------------------------ from tensorflow.compat.v1 import ConfigProto from tensorflow.compat.v1 import InteractiveSession config = ConfigProto() config.gpu_options.allow_growth = T
2020-07-05 16:27:56
759
原创 Tensorflow 2.0 使用tf.data.dataset加载多个.npy文件，以及OOM问题解决
如何在tensorflow2中使用dataset加载文件废了太多脑细胞，实际训练过程中还有好多坑，用作记录 做医学图像训练，图片大小为1024 x 1024并且保存为.npy文件，单个文件大小4M，生成的模拟图片有2000个，全部读取对之后的训练造成很大影响，基本一定会OOM。采用tensorflow推荐的tf.data接口。 参考链接 Tensorflow官方文档 https://www.tensorflow.org/guide/data 加载图片 https://www.tensorflow.org/t
2020-06-10 16:12:20
2115
5
原创 Tensorflow 2.0 Large Model Support安装使用记录-应对OOM
Tensorflow Large Model Support 安装使用 训练1024x1024的图片，GPU内存只有16GB，batch size设定为1也会OOM。搜索之后发现Tensorflow Large Model Support可以自动解决这个问题，自用记录。 下载安装 经过测试后发现只能在Linux上安装，windows不行。安装地址： https://www.ibm.com/support/knowledgecenter/SS5SF7_1.7.0/navigation/wmlce_instal
2020-06-10 11:04:44
668
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人