如何在一个GPU中运行多个tensorflow代码

时间:2017-12-20 07:11:23

标签: tensorflow tensorflow-gpu

由于我的代码占用的GPU内存较少,我想在一个GPU上运行相同代码的多个副本。

我已经设置了os.environ [“CUDA_VISIBLE_DEVICES”] =“0”,并且还设置了

t_config = tf.ConfigProto()
t_config.gpu_options.allow_growth = True
with tf.Session(config=t_config) as sess:

但是当我启动第二个代码时,它将报告“无法创建会话”。我从谷歌搜索,发现大多数人都说在一个GPU中可以使用多个代码。

非常感谢。

2 个答案:

答案 0 :(得分:0)

也许问题出在你的第二个代码上,首先运行第二个代码并测试它是否正常工作,然后相互运行

答案 1 :(得分:0)

我也遇到了同样的问题,解决方案是在执行第二个脚本之前先等待几秒钟。

因此,您的执行顺序应如下所示:

python main.py
sleep 30s; python main.py

不幸的是,我对此没有答案。