当使用全部 4 个 GPU 时，Tensorflow 的 Multi_Gpu_Model 会导致 Ubuntu 18.04 操作系统崩溃答案

【问题标题】：Tensorflow's Multi_Gpu_Model crashes Ubuntu 18.04 OS when using all 4 GPUs当使用全部 4 个 GPU 时，Tensorflow 的 Multi_Gpu_Model 会导致 Ubuntu 18.04 操作系统崩溃
【发布时间】：2019-02-19 14:03:56
【问题描述】：

我正在 Ubuntu 18.04 LTS 桌面上使用 Tensorflow 设置多 GPU 模型。我在 4 NVIDIA RTX 2080 TI 上运行代码并使用 CPU 编译模型。 Windows 10 操作系统上的相同代码功能，如果是 Ubuntu，它会崩溃并且系统会重新启动。我在哪里检查或更改？是操作系统/代码吗？

with tf.device("/cpu:0"):
    model = create_image_model()

# make the model parallel
model = multi_gpu_model(model, gpus=G)

【问题讨论】：

标签： tensorflow nvidia ubuntu-18.04

【解决方案1】：

尝试将tf.ConfigProto(allow_soft_placement=True) 添加到您的会话或估算器配置中。万一这没有帮助，请尝试在 UEFI 中关闭 IOMMU。

https://www.tensorflow.org/guide/using_gpu

【讨论】：