基于tensorflow训练模型的显存不足解决办法
2024-08-31 22:49:29
import tensorflow as tf
import os
os.environ["CUDA_VISIBLE_DEVICES"] = '0' #指定第一块GPU可用
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.5 # 程序最多只能占用指定gpu50%的显存
config.gpu_options.allow_growth = True #程序按需申请内存
sess = tf.Session(config = config)
最新文章
- 总结libevent安装方法
- jQuery性能优化
- Java面试题大全（三）
- easyui combobox 中实现 checkbox
- 如何用Apache POI操作Excel文件-----如何用Apache POI 画一个离散图
- 查看maven项目的依赖关系 mvn dependency:tree
- Function模式 -- 深入理解javascript
- cocos2d-js屏幕任何位置点击开始的实现
- Android开发--ListPreferance 运行报错：android.preference.ListPreference.findIndexOfValue(ListPreference.java:169)
- 11.12 noip模拟试题
- 添加jar
- 解决rsync 同步auth failed on module问题
- Express全系列教程之(七)：cookie的加密
- JSP页面静态包含和动态包含的区别与联系
- shell中的数据生命周期scope
- POJ 3970（最小公倍数LCM）
- Andrew NG 机器学习编程作业3 Octave
- [MS] 微软官网下载安装SQLSERVER2019的rpm
- socket 映射服务器程序
- MT【98】三元对称不等式