在训练模型的过程中遇到程序报錯结果发现时因为电脑显存不够,资源都已经分配完了根据多方查阅,找到了此问题的解决方法在这里记录和分享一下~
在main函数中添加如下这段代码
打开电脑设置->系统->显示->高级显示设置->显示适配器属性
在训练模型的过程中遇到程序报錯结果发现时因为电脑显存不够,资源都已经分配完了根据多方查阅,找到了此问题的解决方法在这里记录和分享一下~
在main函数中添加如下这段代码
打开电脑设置->系统->显示->高级显示设置->显示适配器属性
Tensorflow是啥支持基于cuda内核与cudnn的GPU加速Keras出現较晚,为Tensorflow是啥的高层框架由于Keras使用的方便性与很好的延展性,之后更是作为Tensorflow是啥的官方指定第三方支持开源框架
但两者在使用GPU时都囿一个特点,就是默认为全占满模式在训练的情况下,特别是分步训练时会导致显存溢出导致程序崩溃。
可以使用自适应配置来调整顯存的使用情况
实验室有一块k20c,显存是5gb,我现在需要訓练一个模型,一共有17 M个参数,训练的时候是使用tensorflow是啥官方文档中cifar10的方法把…