python - 如何让tensorflow和theano同时共享gpus?
问题描述
由于一些复杂的原因,我在我的 python 代码中同时使用了 tensorflow 和 theano,并且我有 2 个 gpus 我希望它们共享,但是正如另一个问题中所述,存在一些问题,我想知道是否有一些技巧可以实现这一点,就像指定 tensorflow 只使用 1 个 gpu 而 theano 使用另一个?
现在我只能禁用 theano 的 gpu 使用os.environ['THEANO_FLAGS'] = 'device=cpu,floatX=float64'
,并让 tensorflow 使用所有
os.environ['KERAS_BACKEND'] = 'theano'
os.environ['THEANO_FLAGS'] = 'device=cpu,floatX=float64'
import tensorflow as tf
import keras as ks
解决方案
我没试过这个。但是,如果您有多个 GPU,则可以使用以下技巧强制在 GPU 上运行代码:
import tensorflow as tf
with tf.device('/gpu:0'):
# Run the tensorflow code
import tensorflow as tf
with tf.device('/gpu:1'):
# Run the theano code
希望这可以帮助!
推荐阅读
- python - MS Graph 的授权问题
- javascript - Puppeteer - select based on text children
- excel - 从工作表中的内容限制循环以在 VBA Excel 中匹配和复制
- kubernetes - 如何管理 helm image 标签部署
- javascript - React 的登录重定向错误是否有任何修复?
- arrays - 每次循环在Python中重复时如何从值列表中选择不同的项目
- google-apps-script - 使用脚本格式化 googlesheet 单元格中的文本
- python - 将音乐会 python 数据帧转换为可用于 oracle 批量插入的格式
- git - 事后是否有可能找出合并提交中哪些文件发生冲突?或者一种强制 git 在提交消息中列出文件的方法?
- javafx - 您可以将 GLSL 着色器与 JavaFX 3D 一起使用吗?