- 博客(3)
- 收藏
- 关注
RSS订阅原创 Tensorflow 2.1+Pycharm明明GPU没有被占用却遇到Out of Memory问题
GPU没有被占用时提示Out of Memory记录在开头添加这一段记录在开头添加这一段//#------------------config------------------------from tensorflow.compat.v1 import ConfigProtofrom tensorflow.compat.v1 import InteractiveSessionconfig = ConfigProto()config.gpu_options.allow_growth = T
2020-07-05 16:27:56
661
原创 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
1989
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
600
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人