[深度學習]Ubuntu16.04執行Python程式時會黑屏

我在Ubuntu16.04上呼叫Tensorflow_GPU執行CNN卷積神經網路時候，電腦並沒有發生黑屏的情況，但是最近心血來潮，做了一個GAN的入門模型跑，電腦卻黑屏了，而且顯示卡的風扇聲音很大。谷歌百度了一圈，也沒有找到合適的解決辦法，想要請假下各路大神，有什麼好的辦法嗎？
電腦環境配置:
系統：Ubuntu16.04
GPU: Geforce GTX TITAN
NVIDIA_Driver_Version: 390.87
Cuda_Version: 9.0.176.2
Cudnn_Version: 7.0.5

執行CNN

執行CNN後的結果，並沒有黑屏，所以我認為環境配置應該是沒問題了。

執行GAN

我嘗試了下用Teamviewer去遠端控制程式執行，發現最後程式會卡在這裡，負責執行程式的工作站也黑屏了。可以看到這裡是有計算Loss值的，但是貌似只計算了一次，螢幕就開始黑屏，系統也卡死了。

程式碼段

呼叫GPU的程式段：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.5 # 佔用GPU50%的視訊記憶體
sess = tf.Session(config=config)

卡在這個迴圈了，按道理說應該列印3次輸出的。

for i in range(300):
    z_batch = np.random.normal(0, 1, size=[batch_size, z_dimensions])
    real_image_batch = mnist.train.next_batch(batch_size)[0].reshape([batch_size, 28, 28, 1])
    _, __, dLossReal, dLossFake = sess.run([d_trainer_real, d_trainer_fake, d_loss_real, d_loss_fake],
                                           {x_placeholder: real_image_batch, z_placeholder: z_batch})
    if(i % 100 == 0):
        print("dLossReal:", dLossReal, "dLossFake", dLossFake)